拆镜复刻、爆款开头复刻管理后台完成

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

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from datetime import datetime
 from types import SimpleNamespace
 
 from fastapi import APIRouter, Body, Depends, HTTPException, Path, Query
@@ -293,12 +294,25 @@ async def list_shot_task_sets(
     status: ShotTaskSetStatusEnum | None = Query(None, description="总任务状态筛选：pending_analysis=等待分析，analyzing=分析中，analysis_completed=分析完成，analysis_failed=分析失败，splitting=拆镜中，split_completed=拆镜完成，partial_failed=部分失败，failed=失败，deleted=已软删"),
     analysis_status: ShotAnalysisStatusEnum | None = Query(None, description="原视频分析状态筛选：pending=待分析，processing=分析中，completed=分析完成，failed=分析失败"),
     split_status: ShotSplitStatusEnum | None = Query(None, description="拆镜状态筛选：none=尚未拆镜，pending=待拆镜，processing=拆镜中，completed=拆镜完成，failed=拆镜失败，retry_waiting=等待恢复重试"),
-    keyword: str | None = Query(None, description="标题/原视频内容/分类/受众关键词，模糊搜索；为空不过滤"),
+    keyword: str | None = Query(None, description="标题/原视频内容/分类/受众关键词，模糊搜索；普通用户只在自己的数据内搜索"),
+    user_id: str | None = Query(None, description="管理员专用：按用户ID筛选；普通用户不可使用"),
+    user_name: str | None = Query(None, description="管理员专用：按用户名模糊筛选；普通用户不可使用"),
+    created_start: datetime | None = Query(None, description="管理员专用：创建时间开始，ISO datetime；普通用户不可使用"),
+    created_end: datetime | None = Query(None, description="管理员专用：创建时间结束，ISO datetime；普通用户不可使用"),
     page: int = Query(1, ge=1, description="分页页码，从1开始"),
     page_size: int = Query(20, ge=1, le=100, description="每页数量，范围1-100"),
     current_user: User = Depends(get_current_user),
     db: AsyncSession = Depends(get_db),
 ):
+    admin_only_params = {
+        "user_id": user_id,
+        "user_name": user_name,
+        "created_start": created_start,
+        "created_end": created_end,
+    }
+    if not _safe_user_is_admin(current_user) and any(value is not None and str(value).strip() != "" for value in admin_only_params.values()):
+        raise HTTPException(status_code=403, detail="当前搜索条件仅管理员可用")
+
     return await list_task_sets(
         db,
         current_user=current_user,
@@ -306,6 +320,10 @@ async def list_shot_task_sets(
         analysis_status=analysis_status.value if analysis_status else None,
         split_status=split_status.value if split_status else None,
         keyword=keyword,
+        user_id=user_id,
+        user_name=user_name,
+        created_start=created_start,
+        created_end=created_end,
         page=page,
         page_size=page_size,
     )