feat:导出面板动态编写一半

backend/db/crud.py

@@ -9,7 +9,7 @@ from datetime import datetime, timezone
 from typing import List, Optional
 from sqlalchemy.orm import Session
 
-from .models import MultiAgentTask, UserAgent
+from .models import MultiAgentTask, UserAgent, ExportRecord
 
 
 class MultiAgentTaskCRUD:
@@ -414,3 +414,85 @@ class UserAgentCRUD:
             db.commit()
             db.refresh(agent)
             return agent
+
+
+class ExportRecordCRUD:
+    """导出记录 CRUD 操作"""
+
+    @staticmethod
+    def create(
+        db: Session,
+        task_id: str,
+        user_id: str,
+        export_type: str,
+        file_name: str,
+        file_path: str,
+        file_url: str = "",
+        file_size: int = 0,
+    ) -> ExportRecord:
+        """创建导出记录"""
+        record = ExportRecord(
+            task_id=task_id,
+            user_id=user_id,
+            export_type=export_type,
+            file_name=file_name,
+            file_path=file_path,
+            file_url=file_url,
+            file_size=file_size,
+        )
+        db.add(record)
+        db.commit()
+        db.refresh(record)
+        return record
+
+    @staticmethod
+    def get_by_id(db: Session, record_id: int) -> Optional[ExportRecord]:
+        """根据 ID 获取记录"""
+        return db.query(ExportRecord).filter(ExportRecord.id == record_id).first()
+
+    @staticmethod
+    def get_by_task_id(
+        db: Session, task_id: str, limit: int = 50
+    ) -> List[ExportRecord]:
+        """根据任务 ID 获取导出记录列表"""
+        return (
+            db.query(ExportRecord)
+            .filter(ExportRecord.task_id == task_id)
+            .order_by(ExportRecord.created_at.desc())
+            .limit(limit)
+            .all()
+        )
+
+    @staticmethod
+    def get_by_user_id(
+        db: Session, user_id: str, limit: int = 50
+    ) -> List[ExportRecord]:
+        """根据用户 ID 获取导出记录列表"""
+        return (
+            db.query(ExportRecord)
+            .filter(ExportRecord.user_id == user_id)
+            .order_by(ExportRecord.created_at.desc())
+            .limit(limit)
+            .all()
+        )
+
+    @staticmethod
+    def delete(db: Session, record_id: int) -> bool:
+        """删除导出记录"""
+        record = db.query(ExportRecord).filter(ExportRecord.id == record_id).first()
+        if record:
+            db.delete(record)
+            db.commit()
+            return True
+        return False
+
+    @staticmethod
+    def delete_by_task_id(db: Session, task_id: str) -> bool:
+        """删除任务的所有导出记录"""
+        records = db.query(ExportRecord).filter(ExportRecord.task_id == task_id).all()
+        if records:
+            for record in records:
+                db.delete(record)
+            db.commit()
+            return True
+        return False