会员积分改版V1

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

@@ -34,6 +34,7 @@ CELERY_TASK_IMPORTS = (
     "app.tasks.module_generation_v2_tasks",
     "app.tasks.private_portrait_asset_tasks",
     "app.tasks.celery_runtime_tasks",
+    "app.tasks.credit_tasks",
 )
 
 
@@ -120,6 +121,11 @@ def _beat_schedule() -> dict:
         "schedule": 300,
         "options": {"queue": CeleryQueue.GEN_PRIVATE_PORTRAIT.value},
     }
+    schedule["credit-maintenance-every-minute"] = {
+        "task": CeleryTaskName.CREDIT_MAINTENANCE.value,
+        "schedule": 60,
+        "options": {"queue": CeleryQueue.GEN_CREDIT_MAINTENANCE.value},
+    }
     return schedule
 
 
@@ -181,6 +187,7 @@ if broker_url:
             CeleryTaskName.MODULE_ASYNC_RECOVERY.value: {"ignore_result": True},
             CeleryTaskName.SHOT_ANALYSIS_RECOVERY.value: {"ignore_result": True},
             CeleryTaskName.SHOT_SPLIT_RECOVERY.value: {"ignore_result": True},
+            CeleryTaskName.CREDIT_MAINTENANCE.value: {"ignore_result": True},
         },
         worker_prefetch_multiplier=1,
         worker_cancel_long_running_tasks_on_connection_loss=True,
@@ -240,6 +247,7 @@ if broker_url:
             CeleryTaskName.PRIVATE_PORTRAIT_DELETE_GROUP.value: {"queue": CeleryQueue.GEN_PRIVATE_PORTRAIT.value},
             CeleryTaskName.PRIVATE_PORTRAIT_DELETE_PROJECT.value: {"queue": CeleryQueue.GEN_PRIVATE_PORTRAIT.value},
             CeleryTaskName.PRIVATE_PORTRAIT_RECOVER_REMOTE_DELETES.value: {"queue": CeleryQueue.GEN_PRIVATE_PORTRAIT.value},
+            CeleryTaskName.CREDIT_MAINTENANCE.value: {"queue": CeleryQueue.GEN_CREDIT_MAINTENANCE.value},
         },
     )
 else: