84 lines
3.1 KiB
Python
84 lines
3.1 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
任务输入参数模型模块
|
||
包含任务输入参数相关的数据模型
|
||
"""
|
||
|
||
import enum
|
||
from sqlalchemy import Column, Integer, String, Text, Boolean, Enum, ForeignKey, JSON
|
||
from sqlalchemy.orm import relationship
|
||
from data.models.base import BaseModel
|
||
|
||
class InputParamType(enum.Enum):
|
||
"""
|
||
输入参数类型枚举
|
||
"""
|
||
STRING = 'string' # 字符串
|
||
NUMBER = 'number' # 数字
|
||
BOOLEAN = 'boolean' # 布尔值
|
||
OBJECT = 'object' # 对象
|
||
ARRAY = 'array' # 数组
|
||
NULL = 'null' # 空值
|
||
|
||
class TaskInput(BaseModel):
|
||
"""
|
||
任务输入参数模型
|
||
表示任务执行所需的输入参数
|
||
"""
|
||
__tablename__ = 'task_inputs'
|
||
|
||
task_id = Column(Integer, ForeignKey('tasks.id'), nullable=False, comment='任务ID')
|
||
name = Column(String(100), nullable=False, comment='参数名称')
|
||
param_type = Column(Enum(InputParamType), nullable=False, default=InputParamType.STRING, comment='参数类型')
|
||
description = Column(String(500), nullable=True, comment='参数描述')
|
||
is_required = Column(Boolean, default=False, comment='是否必填')
|
||
default_value = Column(Text, nullable=True, comment='默认值')
|
||
validation_rules = Column(JSON, nullable=True, comment='验证规则（JSON格式）')
|
||
order = Column(Integer, default=0, comment='排序顺序')
|
||
|
||
def __repr__(self):
|
||
return f"<TaskInput(id={self.id}, task_id={self.task_id}, name='{self.name}', type='{self.param_type}')>"
|
||
|
||
@classmethod
|
||
def get_by_task(cls, task_id):
|
||
"""
|
||
获取任务的所有输入参数
|
||
"""
|
||
return cls.query.filter(cls.task_id == task_id, cls.is_deleted == False).order_by(cls.order).all()
|
||
|
||
@classmethod
|
||
def get_by_task_and_name(cls, task_id, name):
|
||
"""
|
||
根据任务ID和参数名称获取输入参数
|
||
"""
|
||
return cls.query.filter(cls.task_id == task_id, cls.name == name, cls.is_deleted == False).first()
|
||
|
||
class TaskInputValue(BaseModel):
|
||
"""
|
||
任务输入参数值模型
|
||
表示任务执行时的实际输入参数值
|
||
"""
|
||
__tablename__ = 'task_input_values'
|
||
|
||
task_record_id = Column(Integer, ForeignKey('task_records.id'), nullable=False, comment='任务记录ID')
|
||
input_id = Column(Integer, ForeignKey('task_inputs.id'), nullable=False, comment='输入参数ID')
|
||
value = Column(Text, nullable=True, comment='参数值')
|
||
|
||
def __repr__(self):
|
||
return f"<TaskInputValue(id={self.id}, task_record_id={self.task_record_id}, input_id={self.input_id})>"
|
||
|
||
@classmethod
|
||
def get_by_record(cls, task_record_id):
|
||
"""
|
||
获取任务记录的所有输入参数值
|
||
"""
|
||
return cls.query.filter(cls.task_record_id == task_record_id, cls.is_deleted == False).all()
|
||
|
||
@classmethod
|
||
def get_by_record_and_input(cls, task_record_id, input_id):
|
||
"""
|
||
根据任务记录ID和输入参数ID获取输入参数值
|
||
"""
|
||
return cls.query.filter(cls.task_record_id == task_record_id, cls.input_id == input_id, cls.is_deleted == False).first() |