Updata2

etl_billiards/orchestration/task_registry.py

@@ -28,6 +28,26 @@ from tasks.check_cutoff_task import CheckCutoffTask
 from tasks.init_dws_schema_task import InitDwsSchemaTask
 from tasks.dws_build_order_summary_task import DwsBuildOrderSummaryTask
 from tasks.data_integrity_task import DataIntegrityTask
+from tasks.seed_dws_config_task import SeedDwsConfigTask
+
+# DWS 层任务导入
+from tasks.dws import (
+    AssistantDailyTask,
+    AssistantMonthlyTask,
+    AssistantCustomerTask,
+    AssistantSalaryTask,
+    AssistantFinanceTask,
+    MemberConsumptionTask,
+    MemberVisitTask,
+    FinanceDailyTask,
+    FinanceRechargeTask,
+    FinanceIncomeStructureTask,
+    FinanceDiscountDetailTask,
+    DwsRetentionCleanupTask,
+    # 指数算法任务
+    RecallIndexTask,
+    IntimacyIndexTask,
+)
 
 class TaskRegistry:
     """任务注册和工厂"""
@@ -81,6 +101,26 @@ default_registry.register("ODS_JSON_ARCHIVE", OdsJsonArchiveTask)
 default_registry.register("CHECK_CUTOFF", CheckCutoffTask)
 default_registry.register("DATA_INTEGRITY_CHECK", DataIntegrityTask)
 default_registry.register("INIT_DWS_SCHEMA", InitDwsSchemaTask)
+default_registry.register("SEED_DWS_CONFIG", SeedDwsConfigTask)
 default_registry.register("DWS_BUILD_ORDER_SUMMARY", DwsBuildOrderSummaryTask)
+
+# DWS 层业务任务
+default_registry.register("DWS_ASSISTANT_DAILY", AssistantDailyTask)
+default_registry.register("DWS_ASSISTANT_MONTHLY", AssistantMonthlyTask)
+default_registry.register("DWS_ASSISTANT_CUSTOMER", AssistantCustomerTask)
+default_registry.register("DWS_ASSISTANT_SALARY", AssistantSalaryTask)
+default_registry.register("DWS_ASSISTANT_FINANCE", AssistantFinanceTask)
+default_registry.register("DWS_MEMBER_CONSUMPTION", MemberConsumptionTask)
+default_registry.register("DWS_MEMBER_VISIT", MemberVisitTask)
+default_registry.register("DWS_FINANCE_DAILY", FinanceDailyTask)
+default_registry.register("DWS_FINANCE_RECHARGE", FinanceRechargeTask)
+default_registry.register("DWS_FINANCE_INCOME_STRUCTURE", FinanceIncomeStructureTask)
+default_registry.register("DWS_FINANCE_DISCOUNT_DETAIL", FinanceDiscountDetailTask)
+default_registry.register("DWS_RETENTION_CLEANUP", DwsRetentionCleanupTask)
+
+# DWS 指数算法任务
+default_registry.register("DWS_RECALL_INDEX", RecallIndexTask)
+default_registry.register("DWS_INTIMACY_INDEX", IntimacyIndexTask)
+
 for code, task_cls in ODS_TASK_CLASSES.items():
     default_registry.register(code, task_cls)