优化脚本

2026-08-25 16:40:22 +08:00
parent e2ec2d14af
commit df6784b90c
22 changed files with 893 additions and 156 deletions
@@ -289,6 +289,12 @@ class ProjectViewSet(TeamScopedViewSetMixin, ModelViewSet):
        group.adopted_asset = tri
        group.save(update_fields=["adopted_asset", "updated_at"])

+    def list(self, request, *args, **kwargs):
+        from .services.quick_create import restore_false_failed_quick_creates
+
+        restore_false_failed_quick_creates(self.get_team())
+        return super().list(request, *args, **kwargs)
+
    def retrieve(self, request, *args, **kwargs):
        # 详情加载(=进入流水线页)时自愈视频片段数:历史项目在「采用前增删分镜」未同步,
        # 或项目创建固定铺的 4 段从未被收口,会让视频步骤的片段数与故事板/采用版分镜对不上。
@@ -699,9 +705,47 @@ class ProjectViewSet(TeamScopedViewSetMixin, ModelViewSet):
        job = self._quick_job_queryset().get(id=job.id)
        return Response(QuickCreateJobSerializer(job).data)

+    @action(detail=False, methods=["post"], url_path=r"quick-create-retry/(?P<job_id>[^/.]+)")
+    def quick_create_retry(self, request, job_id=None):
+        job = self._quick_job_queryset().filter(id=job_id).first()
+        if job is None:
+            return Response({"detail": "极速成片任务不存在"}, status=status.HTTP_404_NOT_FOUND)
+        if job.status == QuickCreateJob.Status.SUCCEEDED:
+            return Response({"detail": "任务已完成"}, status=status.HTTP_400_BAD_REQUEST)
+        if job.status == QuickCreateJob.Status.CANCELLED:
+            return Response({"detail": "已取消的任务请重新开始"}, status=status.HTTP_400_BAD_REQUEST)
+        from .services.quick_create import resume_quick_create
+
+        resume_quick_create(job)
+        job = self._quick_job_queryset().get(id=job.id)
+        return Response(QuickCreateJobSerializer(job).data)
+
+    @action(detail=False, methods=["post"], url_path=r"quick-create-retry/(?P<job_id>[^/.]+)")
+    def quick_create_retry(self, request, job_id=None):
+        job = self._quick_job_queryset().filter(id=job_id).first()
+        if job is None:
+            return Response({"detail": "极速成片任务不存在"}, status=status.HTTP_404_NOT_FOUND)
+        if job.status == QuickCreateJob.Status.SUCCEEDED:
+            return Response({"detail": "任务已经完成"}, status=status.HTTP_400_BAD_REQUEST)
+        if job.status == QuickCreateJob.Status.CANCELLED:
+            return Response({"detail": "已取消的任务请重新开始"}, status=status.HTTP_400_BAD_REQUEST)
+        from .services.quick_create import resume_quick_create
+
+        resume_quick_create(job)
+        job = self._quick_job_queryset().get(id=job.id)
+        return Response(QuickCreateJobSerializer(job).data)
+
    @action(detail=False, methods=["get"], url_path="quick-create-history")
    def quick_create_history(self, request):
-        jobs = self._quick_job_queryset().filter(status=QuickCreateJob.Status.SUCCEEDED).order_by("-created_at")
+        from .services.quick_create import restore_false_failed_quick_creates
+
+        restore_false_failed_quick_creates(self.get_team())
+        # 进行中的任务看上方状态卡；列表要能找回失败后去专业模式继续的项目。
+        jobs = (
+            self._quick_job_queryset()
+            .exclude(status__in=[QuickCreateJob.Status.QUEUED, QuickCreateJob.Status.RUNNING])
+            .order_by("-created_at")
+        )
        return Response({
            "count": jobs.count(),
            "results": QuickCreateJobSerializer(jobs[:30], many=True).data,