Build lightweight AI agent admin
This commit is contained in:
Codex
@@ -0,0 +1,274 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Scheduler Schema - 定时任务数据验证和序列化
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional, List
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
||||
|
||||
from app.base_schema import CSTDatetime
|
||||
|
||||
|
||||
# ==================== SchedulerJob Schemas ====================
|
||||
|
||||
class SchedulerJobBase(BaseModel):
|
||||
"""定时任务基础Schema"""
|
||||
application_id: Optional[str] = Field(None, description="所属应用ID")
|
||||
name: str = Field(..., min_length=1, max_length=128, description="任务名称")
|
||||
code: str = Field(..., min_length=1, max_length=128, description="任务编码")
|
||||
description: Optional[str] = Field(None, description="任务描述")
|
||||
group: str = Field(default="default", max_length=64, description="任务分组")
|
||||
trigger_type: str = Field(..., description="触发器类型：cron/interval/date")
|
||||
cron_expression: Optional[str] = Field(None, max_length=128, description="Cron表达式")
|
||||
interval_seconds: Optional[int] = Field(None, ge=1, description="间隔时间（秒）")
|
||||
run_date: Optional[datetime] = Field(None, description="指定执行时间")
|
||||
task_func: str = Field(..., max_length=256, description="任务函数路径")
|
||||
task_args: Optional[str] = Field(None, description="任务位置参数（JSON）")
|
||||
task_kwargs: Optional[str] = Field(None, description="任务关键字参数（JSON）")
|
||||
status: int = Field(default=0, description="任务状态：0-禁用，1-启用，2-暂停")
|
||||
priority: int = Field(default=0, description="任务优先级")
|
||||
max_instances: int = Field(default=1, ge=1, description="最大实例数")
|
||||
max_retries: int = Field(default=0, ge=0, description="错误重试次数")
|
||||
timeout: Optional[int] = Field(None, ge=1, description="超时时间（秒）")
|
||||
coalesce: bool = Field(default=True, description="是否合并执行")
|
||||
allow_concurrent: bool = Field(default=False, description="是否允许并发执行")
|
||||
remark: Optional[str] = Field(None, description="备注信息")
|
||||
sort: int = Field(default=0, description="排序")
|
||||
|
||||
@field_validator('trigger_type')
|
||||
@classmethod
|
||||
def validate_trigger_type(cls, v):
|
||||
"""验证触发器类型"""
|
||||
if v not in ['cron', 'interval', 'date']:
|
||||
raise ValueError('触发器类型必须是 cron、interval 或 date')
|
||||
return v
|
||||
|
||||
@field_validator('status')
|
||||
@classmethod
|
||||
def validate_status(cls, v):
|
||||
"""验证状态"""
|
||||
if v not in [0, 1, 2]:
|
||||
raise ValueError('状态必须是 0（禁用）、1（启用）或 2（暂停）')
|
||||
return v
|
||||
|
||||
@field_validator('code')
|
||||
@classmethod
|
||||
def validate_code(cls, v):
|
||||
"""验证任务编码格式"""
|
||||
if not v:
|
||||
raise ValueError('任务编码不能为空')
|
||||
if not v.replace('_', '').isalnum():
|
||||
raise ValueError('任务编码只能包含字母、数字和下划线')
|
||||
return v
|
||||
|
||||
|
||||
class SchedulerJobCreate(SchedulerJobBase):
|
||||
"""定时任务创建Schema"""
|
||||
|
||||
@field_validator('cron_expression')
|
||||
@classmethod
|
||||
def validate_cron_expression(cls, v, info):
|
||||
"""验证 Cron 表达式"""
|
||||
if info.data.get('trigger_type') == 'cron' and not v:
|
||||
raise ValueError('Cron 类型任务必须提供 cron_expression')
|
||||
return v
|
||||
|
||||
@field_validator('interval_seconds')
|
||||
@classmethod
|
||||
def validate_interval_seconds(cls, v, info):
|
||||
"""验证间隔时间"""
|
||||
if info.data.get('trigger_type') == 'interval' and not v:
|
||||
raise ValueError('Interval 类型任务必须提供 interval_seconds')
|
||||
return v
|
||||
|
||||
@field_validator('run_date')
|
||||
@classmethod
|
||||
def validate_run_date(cls, v, info):
|
||||
"""验证指定时间"""
|
||||
if info.data.get('trigger_type') == 'date' and not v:
|
||||
raise ValueError('Date 类型任务必须提供 run_date')
|
||||
return v
|
||||
|
||||
|
||||
class SchedulerJobUpdate(BaseModel):
|
||||
"""定时任务更新Schema - 所有字段可选"""
|
||||
application_id: Optional[str] = Field(None, description="所属应用ID")
|
||||
name: Optional[str] = Field(None, min_length=1, max_length=128, description="任务名称")
|
||||
code: Optional[str] = Field(None, min_length=1, max_length=128, description="任务编码")
|
||||
description: Optional[str] = Field(None, description="任务描述")
|
||||
group: Optional[str] = Field(None, max_length=64, description="任务分组")
|
||||
trigger_type: Optional[str] = Field(None, description="触发器类型")
|
||||
cron_expression: Optional[str] = Field(None, max_length=128, description="Cron表达式")
|
||||
interval_seconds: Optional[int] = Field(None, ge=1, description="间隔时间（秒）")
|
||||
run_date: Optional[datetime] = Field(None, description="指定执行时间")
|
||||
task_func: Optional[str] = Field(None, max_length=256, description="任务函数路径")
|
||||
task_args: Optional[str] = Field(None, description="任务位置参数（JSON）")
|
||||
task_kwargs: Optional[str] = Field(None, description="任务关键字参数（JSON）")
|
||||
status: Optional[int] = Field(None, description="任务状态")
|
||||
priority: Optional[int] = Field(None, description="任务优先级")
|
||||
max_instances: Optional[int] = Field(None, ge=1, description="最大实例数")
|
||||
max_retries: Optional[int] = Field(None, ge=0, description="错误重试次数")
|
||||
timeout: Optional[int] = Field(None, ge=1, description="超时时间（秒）")
|
||||
coalesce: Optional[bool] = Field(None, description="是否合并执行")
|
||||
allow_concurrent: Optional[bool] = Field(None, description="是否允许并发执行")
|
||||
remark: Optional[str] = Field(None, description="备注信息")
|
||||
sort: Optional[int] = Field(None, description="排序")
|
||||
|
||||
|
||||
class SchedulerJobResponse(BaseModel):
|
||||
"""定时任务响应Schema"""
|
||||
id: str
|
||||
application_id: Optional[str] = None
|
||||
name: str
|
||||
code: str
|
||||
description: Optional[str] = None
|
||||
group: str
|
||||
trigger_type: str
|
||||
trigger_type_display: Optional[str] = None
|
||||
cron_expression: Optional[str] = None
|
||||
interval_seconds: Optional[int] = None
|
||||
run_date: Optional[CSTDatetime] = None
|
||||
task_func: str
|
||||
task_args: Optional[str] = None
|
||||
task_kwargs: Optional[str] = None
|
||||
status: int
|
||||
status_display: Optional[str] = None
|
||||
priority: int
|
||||
max_instances: int
|
||||
max_retries: int
|
||||
timeout: Optional[int] = None
|
||||
coalesce: bool
|
||||
allow_concurrent: bool
|
||||
total_run_count: int
|
||||
success_count: int
|
||||
failure_count: int
|
||||
success_rate: Optional[float] = None
|
||||
last_run_time: Optional[CSTDatetime] = None
|
||||
next_run_time: Optional[CSTDatetime] = None
|
||||
last_run_status: Optional[str] = None
|
||||
last_run_result: Optional[str] = None
|
||||
remark: Optional[str] = None
|
||||
sort: int = 0
|
||||
sys_create_datetime: Optional[CSTDatetime] = None
|
||||
sys_update_datetime: Optional[CSTDatetime] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class SchedulerJobSimple(BaseModel):
|
||||
"""定时任务简化输出（用于选择器）"""
|
||||
id: str
|
||||
application_id: Optional[str] = None
|
||||
name: str
|
||||
code: str
|
||||
group: str
|
||||
status: int
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class SchedulerJobBatchDeleteIn(BaseModel):
|
||||
"""批量删除输入"""
|
||||
ids: List[str] = Field(..., description="任务ID列表")
|
||||
|
||||
|
||||
class SchedulerJobBatchDeleteOut(BaseModel):
|
||||
"""批量删除输出"""
|
||||
count: int = Field(..., description="删除成功数量")
|
||||
failed_ids: List[str] = Field(default=[], description="删除失败的ID列表")
|
||||
|
||||
|
||||
class SchedulerJobBatchUpdateStatusIn(BaseModel):
|
||||
"""批量更新状态输入"""
|
||||
ids: List[str] = Field(..., description="任务ID列表")
|
||||
status: int = Field(..., description="目标状态：0-禁用，1-启用，2-暂停")
|
||||
|
||||
|
||||
class SchedulerJobBatchUpdateStatusOut(BaseModel):
|
||||
"""批量更新状态输出"""
|
||||
count: int = Field(..., description="更新成功数量")
|
||||
|
||||
|
||||
class SchedulerJobExecuteIn(BaseModel):
|
||||
"""立即执行任务输入"""
|
||||
job_id: str = Field(..., description="任务ID")
|
||||
|
||||
|
||||
class SchedulerJobExecuteOut(BaseModel):
|
||||
"""立即执行任务输出"""
|
||||
success: bool = Field(..., description="是否成功")
|
||||
message: str = Field(..., description="消息")
|
||||
log_id: Optional[str] = Field(None, description="日志ID")
|
||||
|
||||
|
||||
class SchedulerJobStatisticsOut(BaseModel):
|
||||
"""任务统计输出"""
|
||||
total_jobs: int = Field(..., description="总任务数")
|
||||
enabled_jobs: int = Field(..., description="启用任务数")
|
||||
disabled_jobs: int = Field(..., description="禁用任务数")
|
||||
paused_jobs: int = Field(..., description="暂停任务数")
|
||||
total_executions: int = Field(..., description="总执行次数")
|
||||
success_executions: int = Field(..., description="成功执行次数")
|
||||
failed_executions: int = Field(..., description="失败执行次数")
|
||||
success_rate: float = Field(..., description="成功率")
|
||||
|
||||
|
||||
class SchedulerJobSearchRequest(BaseModel):
|
||||
"""搜索任务请求"""
|
||||
keyword: str = Field(..., description="搜索关键词")
|
||||
|
||||
|
||||
# ==================== SchedulerLog Schemas ====================
|
||||
|
||||
class SchedulerLogResponse(BaseModel):
|
||||
"""定时任务日志响应Schema"""
|
||||
id: str
|
||||
job_id: str
|
||||
job_name: str
|
||||
job_code: str
|
||||
status: str
|
||||
status_display: Optional[str] = None
|
||||
start_time: CSTDatetime
|
||||
end_time: Optional[CSTDatetime] = None
|
||||
duration: Optional[float] = None
|
||||
result: Optional[str] = None
|
||||
exception: Optional[str] = None
|
||||
traceback: Optional[str] = None
|
||||
hostname: Optional[str] = None
|
||||
process_id: Optional[int] = None
|
||||
retry_count: int
|
||||
sys_create_datetime: Optional[CSTDatetime] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class SchedulerLogBatchDeleteIn(BaseModel):
|
||||
"""批量删除日志输入"""
|
||||
ids: List[str] = Field(..., description="日志ID列表")
|
||||
|
||||
|
||||
class SchedulerLogBatchDeleteOut(BaseModel):
|
||||
"""批量删除日志输出"""
|
||||
count: int = Field(..., description="删除成功数量")
|
||||
|
||||
|
||||
class SchedulerLogCleanIn(BaseModel):
|
||||
"""清理日志输入"""
|
||||
days: int = Field(..., ge=1, description="保留最近N天的日志")
|
||||
status: Optional[str] = Field(None, description="只清理指定状态的日志")
|
||||
|
||||
|
||||
class SchedulerLogCleanOut(BaseModel):
|
||||
"""清理日志输出"""
|
||||
count: int = Field(..., description="清理数量")
|
||||
|
||||
|
||||
# ==================== Scheduler Status Schemas ====================
|
||||
|
||||
class SchedulerStatusOut(BaseModel):
|
||||
"""调度器状态输出"""
|
||||
is_running: bool = Field(..., description="是否运行中")
|
||||
job_count: int = Field(..., description="任务数量")
|
||||
jobs: List[dict] = Field(default=[], description="任务列表")
|
||||
Reference in New Issue
Block a user