tianfeng_task_modules/data/models/task_backup.py

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

"""
任务备份模型模块
包含任务备份相关的数据模型
"""

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

class BackupType(enum.Enum):
    """
    备份类型枚举
    """
    MANUAL = 'manual'  # 手动备份
    AUTO = 'auto'  # 自动备份
    SYSTEM = 'system'  # 系统备份（如发布前自动备份）

class TaskBackup(BaseModel):
    """
    任务备份模型
    表示任务配置的备份版本
    """
    __tablename__ = 'task_backups'

    task_id = Column(Integer, ForeignKey('tasks.id'), nullable=False, comment='任务ID')
    version_id = Column(Integer, ForeignKey('task_versions.id'), nullable=True, comment='任务版本ID')
    backup_type = Column(Enum(BackupType), nullable=False, default=BackupType.MANUAL, comment='备份类型')
    name = Column(String(100), nullable=True, comment='备份名称')
    description = Column(String(500), nullable=True, comment='备份描述')
    backup_time = Column(DateTime, nullable=False, comment='备份时间')
    created_by = Column(String(100), nullable=True, comment='创建用户ID')
    content = Column(JSON, nullable=False, comment='备份内容（JSON格式）')
    flow_data = Column(JSON, nullable=True, comment='流程数据（JSON格式）')
    is_restorable = Column(Boolean, default=True, comment='是否可恢复')
    
    def __repr__(self):
        return f"<TaskBackup(id={self.id}, task_id={self.task_id}, backup_type='{self.backup_type}')>"
    
    @classmethod
    def get_by_task(cls, task_id, page=1, per_page=20):
        """
        获取任务的所有备份
        """
        return cls.query.filter(
            cls.task_id == task_id,
            cls.is_deleted == False
        ).order_by(cls.backup_time.desc()).paginate(page=page, per_page=per_page)
    
    @classmethod
    def create_backup(cls, task_id, version_id, content, flow_data=None, backup_type=BackupType.MANUAL, 
                     name=None, description=None, created_by=None):
        """
        创建任务备份
        """
        import datetime
        from config.database import db_session
        
        backup = cls(
            task_id=task_id,
            version_id=version_id,
            backup_type=backup_type,
            name=name,
            description=description,
            backup_time=datetime.datetime.now(),
            created_by=created_by,
            content=content,
            flow_data=flow_data,
            is_restorable=True
        )
        
        db_session.add(backup)
        db_session.commit()
        
        # 记录用户操作
        from data.models.user_operation import UserOperation, OperationType, TargetType
        if created_by:
            UserOperation.log_operation(
                user_id=created_by,
                user_name=None,  # 用户名称需要从外部系统获取
                operation_type=OperationType.CREATE,
                target_type=TargetType.TASK,
                target_id=task_id,
                target_name=name or f"Backup {backup.id}",
                operation_details={
                    'backup_id': backup.id,
                    'backup_type': backup_type.value,
                    'version_id': version_id
                }
            )
        
        return backup
    
    @classmethod
    def restore_backup(cls, backup_id, restored_by=None):
        """
        恢复任务备份
        """
        from config.database import db_session
        
        backup = cls.query.get(backup_id)
        if not backup or backup.is_deleted or not backup.is_restorable:
            return None
        
        # 获取任务和任务版本
        from data.models.task import Task, TaskVersion
        task = Task.query.get(backup.task_id)
        if not task or task.is_deleted:
            return None
        
        # 创建新版本
        version = TaskVersion(
            task_id=task.id,
            version=f"{float(task.current_version.version) + 0.1:.1f}" if task.current_version else "1.0",
            content=backup.content,
            flow_data=backup.flow_data,
            remark=f"Restored from backup {backup.id}"
        )
        
        db_session.add(version)
        db_session.commit()
        
        # 更新任务的当前版本
        task.current_version_id = version.id
        task.updated_by = restored_by
        db_session.commit()
        
        # 记录用户操作
        from data.models.user_operation import UserOperation, OperationType, TargetType
        if restored_by:
            UserOperation.log_operation(
                user_id=restored_by,
                user_name=None,  # 用户名称需要从外部系统获取
                operation_type=OperationType.UPDATE,
                target_type=TargetType.TASK,
                target_id=task.id,
                target_name=task.name,
                operation_details={
                    'backup_id': backup.id,
                    'version_id': version.id,
                    'action': 'restore'
                }
            )
        
        return version