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, ExportRecord
+from .models import MultiAgentTask, UserAgent, ExportRecord, PlanShare
 
 
 class MultiAgentTaskCRUD:
@@ -500,3 +500,78 @@ class ExportRecordCRUD:
             db.commit()
             return True
         return False
+
+
+class PlanShareCRUD:
+    """任务分享 CRUD 操作"""
+
+    @staticmethod
+    def create(
+        db: Session,
+        share_token: str,
+        task_id: str,
+        task_data: dict,
+        expires_at: Optional[datetime] = None,
+        extraction_code: Optional[str] = None,
+    ) -> PlanShare:
+        """创建分享记录"""
+        share = PlanShare(
+            share_token=share_token,
+            extraction_code=extraction_code,
+            task_id=task_id,
+            task_data=task_data,
+            expires_at=expires_at,
+        )
+        db.add(share)
+        db.commit()
+        db.refresh(share)
+        return share
+
+    @staticmethod
+    def get_by_token(db: Session, share_token: str) -> Optional[PlanShare]:
+        """根据 token 获取分享记录"""
+        return db.query(PlanShare).filter(PlanShare.share_token == share_token).first()
+
+    @staticmethod
+    def get_by_task_id(
+        db: Session, task_id: str, limit: int = 10
+    ) -> List[PlanShare]:
+        """根据任务 ID 获取分享记录列表"""
+        return (
+            db.query(PlanShare)
+            .filter(PlanShare.task_id == task_id)
+            .order_by(PlanShare.created_at.desc())
+            .limit(limit)
+            .all()
+        )
+
+    @staticmethod
+    def increment_view_count(db: Session, share_token: str) -> Optional[PlanShare]:
+        """增加查看次数"""
+        share = db.query(PlanShare).filter(PlanShare.share_token == share_token).first()
+        if share:
+            share.view_count = (share.view_count or 0) + 1
+            db.commit()
+            db.refresh(share)
+        return share
+
+    @staticmethod
+    def delete(db: Session, share_token: str) -> bool:
+        """删除分享记录"""
+        share = db.query(PlanShare).filter(PlanShare.share_token == share_token).first()
+        if share:
+            db.delete(share)
+            db.commit()
+            return True
+        return False
+
+    @staticmethod
+    def delete_by_task_id(db: Session, task_id: str) -> bool:
+        """删除任务的所有分享记录"""
+        shares = db.query(PlanShare).filter(PlanShare.task_id == task_id).all()
+        if shares:
+            for share in shares:
+                db.delete(share)
+            db.commit()
+            return True
+        return False