生成项目任务/chat任务软删|生成资源管控回收|生成资源token验签API预处理

video-gen-api/app/api/v1/generation_ai.py

@@ -1,3 +1,5 @@
+from datetime import datetime, timezone
+
 from fastapi import APIRouter, Body, Depends, HTTPException, Path, Query
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -10,6 +12,7 @@ from app.schemas.generation_ai import (
     GenerationAIHistoryDayItemsOut,
     GenerationAIHistoryGroupedOut,
     GenerationAIRetryOut,
+    GenerationAITaskDeleteOut,
     GenerationAITaskCreate,
     GenerationAITaskListOut,
     GenerationAITaskOut,
@@ -21,6 +24,7 @@ from app.services.generation_ai_service import (
     list_generation_history_day_items,
     list_generation_history_grouped_days,
     record_to_out,
+    soft_delete_chat_generation_task,
 )
 from app.services.generation_log_service import log_task_event
 from app.tasks.celery_app import celery_app
@@ -389,6 +393,7 @@ async def get_task(
             ChatGenerationTask.id == task_id,
             ChatGenerationTask.user_id == current_user.id,
             ChatGenerationTask.generation_mode == "chatapi_async",
+            ChatGenerationTask.deleted_at.is_(None),
         )
     )
     task = result.scalar_one_or_none()
@@ -397,6 +402,71 @@ async def get_task(
     return record_to_out(task)
 
 
+@router.delete(
+    "/tasks/{task_id}",
+    response_model=GenerationAITaskDeleteOut,
+    summary="删除AI生成任务",
+    description=(
+        "软删除当前登录用户自己的AI生成任务。"
+        "该接口不会物理删除数据库记录和本地文件，只会设置 deleted_at，后续列表、详情、历史统计默认不再返回。"
+        "删除已完成任务时会联动软删 generated_resources 资源账本，并重新扣减用户有效资源空间统计。"
+        "如果任务仍处于 generating 生成中状态，接口会直接拦截，不允许删除。"
+    ),
+    responses={
+        200: {
+            "description": "软删除成功，返回任务ID和本次释放的资源空间字节数",
+        },
+        400: {
+            "description": "任务正在生成中，暂不能删除",
+        },
+        401: {
+            "description": "未登录或 Token 无效",
+        },
+        404: {
+            "description": "任务不存在，或任务不属于当前用户，或任务已经被删除",
+        },
+    },
+)
+async def delete_task(
+    task_id: str = Path(
+        ...,
+        description="需要删除的AI生成任务ID",
+        examples=["0019e0a44895b6d837d"],
+    ),
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    result = await db.execute(
+        select(ChatGenerationTask).where(
+            ChatGenerationTask.id == task_id,
+            ChatGenerationTask.user_id == current_user.id,
+            ChatGenerationTask.generation_mode == "chatapi_async",
+            ChatGenerationTask.deleted_at.is_(None),
+        )
+    )
+    task = result.scalar_one_or_none()
+    if not task:
+        raise HTTPException(status_code=404, detail="任务不存在")
+
+    if task.status == "generating":
+        raise HTTPException(status_code=400, detail="当前任务正在生成中，暂不能删除")
+
+    deleted_at = datetime.now(timezone.utc)
+    freed_size_bytes = await soft_delete_chat_generation_task(
+        db,
+        task=task,
+        deleted_at=deleted_at,
+    )
+    await db.flush()
+
+    return GenerationAITaskDeleteOut(
+        message="任务已删除",
+        task_id=task.id,
+        deleted=True,
+        freed_size_bytes=freed_size_bytes,
+    )
+
+
 @router.post(
     "/tasks/{task_id}/retry",
     response_model=GenerationAIRetryOut,
@@ -443,6 +513,7 @@ async def retry_task(
             ChatGenerationTask.id == task_id,
             ChatGenerationTask.user_id == current_user.id,
             ChatGenerationTask.generation_mode == "chatapi_async",
+            ChatGenerationTask.deleted_at.is_(None),
         )
     )
     task = result.scalar_one_or_none()