60 lines
2.8 KiB
Python
60 lines
2.8 KiB
Python
"""
|
||
任务实例模型
|
||
用于存储任务管理表的数据，记录每次编辑任务的实例
|
||
"""
|
||
from sqlalchemy import Column, Integer, String, Text, DateTime, Boolean, ForeignKey, JSON, Enum
|
||
from sqlalchemy.orm import relationship
|
||
from datetime import datetime
|
||
import enum
|
||
import uuid
|
||
from data.models.base import BaseModel
|
||
|
||
class TaskInstanceStatus(enum.Enum):
|
||
"""任务实例状态枚举"""
|
||
EDITING = "editing" # 编辑中
|
||
PUBLISHED = "published" # 已发布
|
||
|
||
class TaskInstance(BaseModel):
|
||
"""任务实例模型"""
|
||
__tablename__ = "task_instances"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment="主键ID")
|
||
instance_id = Column(String(36), nullable=False, unique=True, index=True, comment="实例唯一ID，用于外部引用")
|
||
task_id = Column(String(50), ForeignKey("tasks.task_id"), nullable=False, comment="关联的任务ID")
|
||
name = Column(String(100), nullable=False, comment="任务名称（复制自任务表）")
|
||
variables = Column(JSON, nullable=True, comment="任务变量")
|
||
priority = Column(Integer, default=1, comment="任务优先级")
|
||
input_params = Column(JSON, nullable=True, comment="任务输入参数")
|
||
block_outputs = Column(JSON, nullable=True, comment="块输出参数")
|
||
context_params = Column(JSON, nullable=True, comment="上下文参数")
|
||
status = Column(Enum(TaskInstanceStatus), default=TaskInstanceStatus.EDITING, comment="任务实例状态")
|
||
created_at = Column(DateTime, default=datetime.now, comment="创建时间")
|
||
updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now, comment="更新时间")
|
||
is_deleted = Column(Boolean, default=False, comment="是否删除")
|
||
|
||
# 关联关系
|
||
task = relationship("Task", back_populates="instances")
|
||
|
||
def __init__(self, **kwargs):
|
||
"""初始化实例，自动生成instance_id"""
|
||
if 'instance_id' not in kwargs:
|
||
kwargs['instance_id'] = str(uuid.uuid4())
|
||
super(TaskInstance, self).__init__(**kwargs)
|
||
|
||
def to_dict(self):
|
||
"""转换为字典"""
|
||
return {
|
||
"id": self.instance_id, # 使用instance_id作为对外ID
|
||
"instance_id": self.instance_id,
|
||
"task_id": self.task_id,
|
||
"name": self.name,
|
||
"variables": self.variables or {},
|
||
"priority": self.priority,
|
||
"input_params": self.input_params or {},
|
||
"block_outputs": self.block_outputs or {},
|
||
"context_params": self.context_params or {},
|
||
"status": self.status.value if self.status else None,
|
||
"created_at": int(self.created_at.timestamp() * 1000) if self.created_at else None,
|
||
"updated_at": int(self.updated_at.timestamp() * 1000) if self.updated_at else None,
|
||
"is_deleted": self.is_deleted
|
||
} |