112 lines
4.5 KiB
Python
112 lines
4.5 KiB
Python
"""
|
||
任务输入参数模型
|
||
用于存储任务输入参数的数据
|
||
"""
|
||
from sqlalchemy import Column, Integer, String, Text, DateTime, Boolean, ForeignKey, JSON, Enum, Index
|
||
from sqlalchemy.orm import relationship, foreign
|
||
from sqlalchemy.sql.expression import and_
|
||
from datetime import datetime
|
||
import uuid
|
||
import json
|
||
from data.models.base import BaseModel
|
||
from config.task_config import TaskInputParamType
|
||
|
||
class TaskInputParam(BaseModel):
|
||
"""任务输入参数模型"""
|
||
__tablename__ = "task_input_params"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True, comment="主键ID")
|
||
param_id = Column(String(36), nullable=False, unique=True, index=True, comment="参数唯一ID，用于外部引用")
|
||
instance_id = Column(String(36), index=True, nullable=False, comment="关联的任务实例ID")
|
||
task_id = Column(String(50), nullable=False, index=True, comment="任务ID，冗余存储便于查询")
|
||
param_name = Column(String(50), nullable=True, comment="参数名称，变量名")
|
||
label = Column(String(100), nullable=True, comment="参数标签，显示名称")
|
||
param_type = Column(String(20), nullable=False, comment="参数类型")
|
||
required = Column(Boolean, default=False, comment="是否必填")
|
||
_default_value = Column('default_value', Text, nullable=True, comment="默认值")
|
||
description = Column(Text, nullable=True, comment="参数说明")
|
||
is_system = Column(Boolean, default=False, comment="是否系统参数")
|
||
is_readonly = Column(Boolean, default=False, comment="是否只读参数")
|
||
sort_order = Column(Integer, default=0, comment="排序顺序")
|
||
|
||
# 手动创建索引而不是使用外键约束
|
||
__table_args__ = (
|
||
Index('idx_task_input_params_instance_id', 'instance_id'),
|
||
{'mysql_engine': 'InnoDB'}
|
||
)
|
||
|
||
# 关联关系 - 使用foreign()显式标记外键
|
||
instance = relationship(
|
||
"TaskInstance",
|
||
primaryjoin="and_(TaskInputParam.instance_id==foreign(TaskInstance.instance_id), "
|
||
"TaskInstance.is_deleted==False)",
|
||
foreign_keys=[instance_id],
|
||
back_populates="input_params",
|
||
viewonly=True
|
||
)
|
||
|
||
def __init__(self, **kwargs):
|
||
"""初始化实例，自动生成param_id"""
|
||
if 'param_id' not in kwargs:
|
||
kwargs['param_id'] = str(uuid.uuid4())
|
||
|
||
# 处理default_value参数
|
||
if 'default_value' in kwargs:
|
||
self.default_value = kwargs.pop('default_value')
|
||
|
||
super(TaskInputParam, self).__init__(**kwargs)
|
||
|
||
@property
|
||
def default_value(self):
|
||
"""获取默认值"""
|
||
if self._default_value is None:
|
||
return None
|
||
|
||
try:
|
||
return json.loads(self._default_value)
|
||
except (json.JSONDecodeError, TypeError):
|
||
return self._default_value
|
||
|
||
@default_value.setter
|
||
def default_value(self, value):
|
||
"""设置默认值"""
|
||
if value is None:
|
||
self._default_value = None
|
||
return
|
||
|
||
if isinstance(value, (str, int, float, bool)):
|
||
# 基本类型直接转为JSON字符串
|
||
self._default_value = json.dumps(value)
|
||
elif isinstance(value, (list, dict)):
|
||
# 复杂类型先验证是否可序列化，然后转为JSON字符串
|
||
try:
|
||
self._default_value = json.dumps(value)
|
||
except:
|
||
self._default_value = None
|
||
else:
|
||
# 其他类型尝试序列化，失败则设为None
|
||
try:
|
||
self._default_value = json.dumps(value)
|
||
except:
|
||
self._default_value = None
|
||
|
||
def to_dict(self):
|
||
"""转换为字典"""
|
||
return {
|
||
"id": self.param_id, # 使用param_id作为对外ID
|
||
"param_id": self.param_id,
|
||
"instance_id": self.instance_id,
|
||
"task_id": self.task_id,
|
||
"param_name": self.param_name,
|
||
"label": self.label,
|
||
"param_type": self.param_type,
|
||
"required": self.required,
|
||
"default_value": self.default_value,
|
||
"description": self.description,
|
||
"is_system": self.is_system,
|
||
"is_readonly": self.is_readonly,
|
||
"sort_order": self.sort_order,
|
||
"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
|
||
} |