交易流水对账明细导出完成

video-gen-api/app/tasks/shot_replicate_tasks.py

@@ -7,6 +7,7 @@ from typing import Any
 from sqlalchemy import select
 
 from app.config import settings
+from app.enums.credit_record import CreditRecordBillingScene, CreditRecordOwnerType
 from app.enums.shot_replicate import (
     ModuleCodeEnum,
     ShotAnalysisStatusEnum,
@@ -36,6 +37,7 @@ from app.services.module_async_recovery_service import (
 )
 from app.services.shot_replicate_taskset_service import refresh_task_set_split_summary
 from app.services.shot_video_analysis_service import analyze_video_for_shot_split
+from app.services.generation_billing_service import charge_shot_video_analysis_usage
 from app.services.shot_video_split_service import split_video_segment_async
 from app.services.upload_video_asset_service import validate_split_range
 from app.tasks.async_runner import run_async
@@ -141,6 +143,16 @@ async def _run_analyze_original_video(task_set_id: str) -> None:
             task_set.analysis_status = ShotAnalysisStatusEnum.COMPLETED.value
             task_set.status = ShotTaskSetStatusEnum.ANALYSIS_COMPLETED.value
             task_set.analysis_error_message = None
+            await charge_shot_video_analysis_usage(
+                db,
+                user_id=task_set.user_id,
+                owner_type=CreditRecordOwnerType.SHOT_REPLICATE_TASK_SET.value,
+                owner_id=task_set.id,
+                usage=analyzed.usage,
+                description="拆镜复刻-原视频分析",
+                billing_scene=CreditRecordBillingScene.SHOT_ORIGINAL_VIDEO_ANALYSIS.value,
+                source_project_id=task_set.id,
+            )
             await db.commit()
             await cleanup_active_if_terminal(db, object_type=OBJECT_SHOT_TASK_SET_ANALYSIS, object_id=task_set_id)
 
@@ -260,6 +272,17 @@ async def _run_analyze_custom_segment_video(segment_id: str) -> None:
             segment.analysis_json = result_json
             segment.analysis_status = ShotSegmentAnalysisStatusEnum.COMPLETED.value
             segment.analysis_error_message = None
+            await charge_shot_video_analysis_usage(
+                db,
+                user_id=segment.user_id,
+                owner_type=CreditRecordOwnerType.SHOT_REPLICATE_SEGMENT.value,
+                owner_id=segment.id,
+                usage=analyzed.usage,
+                description="拆镜复刻-片段视频分析",
+                billing_scene=CreditRecordBillingScene.SHOT_SEGMENT_VIDEO_ANALYSIS.value,
+                source_project_id=segment.task_set_id,
+                source_step_id=segment.id,
+            )
             await db.commit()
             await cleanup_active_if_terminal(db, object_type=OBJECT_SHOT_SEGMENT_ANALYSIS, object_id=segment_id)