datax-admin/app/schemas/jm_job_info.py

from typing import List, Optional

from pydantic import BaseModel

from app.schemas.jm_job_node import JmJobEdge, JmJobNode

class JmJobInfoBase(BaseModel):
    # 任务名称
    name: str
    # 任务类型
    type: str
    # 任务分类
    tag: str
    # 周期类型(1:单次;2:循环)
    cron_type: int
    # 周期num
    cron_num: Optional[int]
    # 周期单位
    cron_unit: Optional[str]
    # 图形信息
    json_str : Optional[str]
    # 用户id
    user_id: str
    # 用户名称
    user_name: str
    # 项目id
    project_id: str


class JmJobInfoCreate(JmJobInfoBase):
    nodes: List[JmJobNode]
    edges: Optional[List[JmJobEdge]] = None
    class Config:
        schema_extra = {
            "example": {
                "name": "example",
                "type": "单作业离线任务",
                "tag": "业务预测",
                "cron_type": "2",
                "cron_num": "1",
                "cron_unit": "时",
                "json_str": "\{图形信息\}",
                "user_id": "test",
                "user_name": "test",
                "project_id": "test",
                "nodes": [
                    {
                        "id": 1,
                        "homework_id": 0,
                        "homework_name": "开始",
                        "start_point": 0,
                    },
                    {
                        "id": 2,
                        "homework_id": 1,
                        "homework_name": "test",
                        "start_point": 1,
                    }
                ],
                "edges": [
                    {
                        "source": 1,
                        "target": 2,
                    }
                ],
            }
        }

class JmJobInfoUpdate(JmJobInfoBase):
    id: int
    nodes: List[JmJobNode]
    edges: Optional[List[JmJobEdge]] = None
    class Config:
        schema_extra = {
            "example": {
                "id": 11,
                "name": "example",
                "type": "单作业离线任务",
                "tag": "业务预测",
                "cron_type": "2",
                "cron_num": "1",
                "cron_unit": "时",
                "json_str": "\{图形信息\}",
                "user_id": "test",
                "user_name": "test",
                "project_id": "test",
                "nodes": [
                    {
                        "id": 1,
                        "homework_id": 0,
                        "homework_name": "开始",
                        "start_point": 0,
                    },
                    {
                        "id": 2,
                        "homework_id": 1,
                        "homework_name": "test",
                        "start_point": 1,
                    }
                ],
                "edges": [
                    {
                        "source": 1,
                        "target": 2,
                    }
                ],
            }
        }

class JmJobInfoStatusUpdate(BaseModel):
    id: int
    status: int
    class Config:
        schema_extra = {
            "example": {
                "id": 1,
                "status": 0,
            }
        }