MoviePilot/app/schemas/workflow.py

from typing import Optional, List, Tuple

from pydantic import BaseModel, Field

from app.schemas import Context, MediaInfo, FileItem, Site, Subscribe, Notification, DownloadTask


class Workflow(BaseModel):
    """
    工作流信息
    """
    name: Optional[str] = Field(None, description="工作流名称")
    description: Optional[str] = Field(None, description="工作流描述")
    timer: Optional[str] = Field(None, description="定时器")
    state: Optional[str] = Field(None, description="状态")
    current_action: Optional[str] = Field(None, description="当前执行动作")
    result: Optional[str] = Field(None, description="任务执行结果")
    run_count: Optional[int] = Field(0, description="已执行次数")
    actions: Optional[list] = Field([], description="任务列表")
    add_time: Optional[str] = Field(None, description="创建时间")
    last_time: Optional[str] = Field(None, description="最后执行时间")


class Action(BaseModel):
    """
    动作信息
    """
    name: Optional[str] = Field(None, description="动作名称")
    description: Optional[str] = Field(None, description="动作描述")
    loop: Optional[bool] = Field(False, description="是否需要循环")
    loop_interval: Optional[int] = Field(0, description="循环间隔 (秒)")
    params: Optional[dict] = Field({}, description="参数")


class ActionContext(BaseModel):
    """
    动作基础上下文，各动作通用数据
    """
    content: Optional[str] = Field(None, description="文本类内容")
    torrents: Optional[List[Context]] = Field([], description="资源列表")
    medias: Optional[List[MediaInfo]] = Field([], description="媒体列表")
    fileitems: Optional[List[FileItem]] = Field([], description="文件列表")
    downloads: Optional[List[DownloadTask]] = Field([], description="下载任务列表")
    sites: Optional[List[Site]] = Field([], description="站点列表")
    subscribes: Optional[List[Subscribe]] = Field([], description="订阅列表")
    messages: Optional[List[Notification]] = Field([], description="消息列表")


class ActionParams(BaseModel):
    """
    动作基础参数
    """
    pass