拆镜复刻片段重试切片API | 真人/虚拟素材库上传API

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

@@ -7,6 +7,7 @@ from fastapi import APIRouter, Body, Depends, File, HTTPException, Path, Query,
 from sqlalchemy import inspect as sa_inspect
 from sqlalchemy.ext.asyncio import AsyncSession
 
+from app.config import settings
 from app.dependencies import get_current_user, get_db
 from app.models.user import User
 from app.enums.shot_replicate import (
@@ -30,6 +31,8 @@ from app.schemas.shot_replicate import (
     ShotReplicateImagePromptUpdateRequest,
     ShotReanalyzeOut,
     ShotReanalyzeRequest,
+    ShotSegmentSplitRetryOut,
+    ShotSplitRetryRequest,
     ShotReplicateMaterialUpdateRequest,
     ShotReplicateSpecOut,
     ShotReplicateTaskDetailOut,
@@ -71,6 +74,7 @@ from app.services.shot_replicate_taskset_service import (
     list_task_sets,
     prepare_reanalyze_segment,
     prepare_reanalyze_task_set,
+    prepare_retry_split_segment,
     segment_detail,
     task_set_detail,
 )
@@ -685,6 +689,83 @@ async def reanalyze_segment(
     return out
 
 
+@router.post(
+    "/segments/{segment_id}/retry-split",
+    response_model=ShotSegmentSplitRetryOut,
+    summary="重试拆镜片段视频切片",
+    description="用于处理 ShotReplicateSegment 视频切片失败；重置 split_status 后复用现有 split_one_segment Celery 任务重新切割。",
+)
+async def retry_split_segment(
+    segment_id: str = Path(..., description="拆镜片段ID，即 shot_replicate_segments.id"),
+    req: ShotSplitRetryRequest = Body(default_factory=ShotSplitRetryRequest, description="切片失败重试参数"),
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    _ensure_celery_enabled(current_user=current_user, step_id=segment_id)
+    try:
+        out = await prepare_retry_split_segment(
+            db,
+            current_user=current_user,
+            segment_id=segment_id,
+            force=req.force,
+            reason=req.reason,
+        )
+        task_set_id = out.task_set_id
+        await db.commit()
+    except HTTPException:
+        await db.rollback()
+        raise
+    except Exception as exc:
+        await db.rollback()
+        _log_api_error(
+            event_type=ShotReplicateLogEventEnum.SEGMENT_SPLIT_RETRY_DISPATCH_FAILED.value,
+            current_user=current_user,
+            step_id=segment_id,
+            message=f"切片失败重试状态重置失败: {exc}",
+            detail={"segment_id": segment_id, "request": req.model_dump()},
+            exc=exc,
+        )
+        raise HTTPException(status_code=500, detail=f"切片失败重试状态重置失败: {exc}")
+
+    try:
+        from app.tasks.shot_replicate_tasks import split_one_segment
+
+        await register_shot_split_task(segment_id, task_set_id=task_set_id)
+        split_one_segment.apply_async(
+            args=[segment_id],
+            queue="gen_result_download",
+            countdown=0,
+            priority=settings.DOWNLOAD_TASK_PRIORITY_RECOVER,
+        )
+        log_module_event_file(
+            module=MODULE,
+            event_type=ShotReplicateLogEventEnum.SEGMENT_SPLIT_RETRY_SUBMITTED.value,
+            project_id=task_set_id,
+            step_id=segment_id,
+            user_id=_safe_user_id(current_user),
+            message="拆镜片段切片重试任务已投递",
+            detail={
+                "segment_id": segment_id,
+                "task_set_id": task_set_id,
+                "task": "split_one_segment",
+                "queue": "gen_result_download",
+                "request": req.model_dump(),
+            },
+        )
+    except Exception as exc:
+        _log_api_error(
+            event_type=ShotReplicateLogEventEnum.SEGMENT_SPLIT_RETRY_DISPATCH_FAILED.value,
+            current_user=current_user,
+            project_id=task_set_id,
+            step_id=segment_id,
+            message=f"拆镜片段切片重试任务投递失败，等待恢复任务兜底: {exc}",
+            detail={"segment_id": segment_id, "task_set_id": task_set_id, "task": "split_one_segment"},
+            exc=exc,
+        )
+        out.message = "切片状态已重置，但 Celery 投递失败，将等待恢复任务兜底"
+    return out
+
+
 @router.delete(
     "/segments/{segment_id}",
     response_model=ShotSegmentDeleteOut,