#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
子任务模型模块
包含子任务和子任务版本相关的数据模型
"""

from sqlalchemy import Column, Integer, String, Text, Boolean, ForeignKey, JSON
from sqlalchemy.orm import relationship
from data.models.base import BaseModel

class SubTask(BaseModel):
    """
    子任务模型
    表示一个可重用的子任务
    """
    __tablename__ = 'subtasks'

    name = Column(String(100), nullable=False, comment='子任务名称')
    code = Column(String(50), nullable=False, unique=True, comment='子任务编码')
    description = Column(String(500), nullable=True, comment='子任务描述')
    is_system = Column(Boolean, default=False, comment='是否为系统子任务')
    is_enabled = Column(Boolean, default=True, comment='是否启用')
    current_version_id = Column(Integer, nullable=True, comment='当前版本ID')
    
    # 关联关系
    versions = relationship('SubTaskVersion', back_populates='subtask', cascade='all, delete-orphan')
    
    def __repr__(self):
        return f"<SubTask(id={self.id}, name='{self.name}', code='{self.code}')>"
    
    @classmethod
    def get_by_code(cls, code):
        """
        根据编码获取子任务
        """
        return cls.query.filter(cls.code == code, cls.is_deleted == False).first()
    
    @classmethod
    def get_enabled_subtasks(cls):
        """
        获取所有启用的子任务
        """
        return cls.query.filter(cls.is_enabled == True, cls.is_deleted == False).all()
    
    @classmethod
    def get_system_subtasks(cls):
        """
        获取所有系统子任务
        """
        return cls.query.filter(cls.is_system == True, cls.is_enabled == True, cls.is_deleted == False).all()

class SubTaskVersion(BaseModel):
    """
    子任务版本模型
    表示子任务的一个版本
    """
    __tablename__ = 'subtask_versions'

    subtask_id = Column(Integer, ForeignKey('subtasks.id'), nullable=False, comment='子任务ID')
    version = Column(String(20), nullable=False, comment='版本号')
    content = Column(JSON, nullable=True, comment='子任务内容（JSON格式）')
    flow_data = Column(JSON, nullable=True, comment='流程数据（JSON格式）')
    input_schema = Column(JSON, nullable=True, comment='输入参数模式（JSON Schema）')
    output_schema = Column(JSON, nullable=True, comment='输出参数模式（JSON Schema）')
    remark = Column(String(500), nullable=True, comment='版本备注')
    
    # 关联关系
    subtask = relationship('SubTask', back_populates='versions')
    
    def __repr__(self):
        return f"<SubTaskVersion(id={self.id}, subtask_id={self.subtask_id}, version='{self.version}')>"
    
    @classmethod
    def get_latest_version(cls, subtask_id):
        """
        获取子任务的最新版本
        """
        return cls.query.filter(
            cls.subtask_id == subtask_id, 
            cls.is_deleted == False
        ).order_by(cls.created_at.desc()).first()
    
    @classmethod
    def get_by_version(cls, subtask_id, version):
        """
        根据版本号获取子任务版本
        """
        return cls.query.filter(
            cls.subtask_id == subtask_id, 
            cls.version == version, 
            cls.is_deleted == False
        ).first()