合并

etl_billiards/orchestration/task_registry.py

@@ -15,10 +15,19 @@ from tasks.table_discount_task import TableDiscountTask
 from tasks.assistant_abolish_task import AssistantAbolishTask
 from tasks.ledger_task import LedgerTask
 from tasks.ods_tasks import ODS_TASK_CLASSES
-from tasks.ticket_dwd_task import TicketDwdTask
 from tasks.manual_ingest_task import ManualIngestTask
 from tasks.payments_dwd_task import PaymentsDwdTask
 from tasks.members_dwd_task import MembersDwdTask
+from tasks.init_schema_task import InitOdsSchemaTask
+from tasks.init_dwd_schema_task import InitDwdSchemaTask
+from tasks.dwd_load_task import DwdLoadTask
+from tasks.ticket_dwd_task import TicketDwdTask
+from tasks.dwd_quality_task import DwdQualityTask
+from tasks.ods_json_archive_task import OdsJsonArchiveTask
+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
 
 class TaskRegistry:
     """任务注册和工厂"""
@@ -64,5 +73,14 @@ default_registry.register("TICKET_DWD", TicketDwdTask)
 default_registry.register("MANUAL_INGEST", ManualIngestTask)
 default_registry.register("PAYMENTS_DWD", PaymentsDwdTask)
 default_registry.register("MEMBERS_DWD", MembersDwdTask)
+default_registry.register("INIT_ODS_SCHEMA", InitOdsSchemaTask)
+default_registry.register("INIT_DWD_SCHEMA", InitDwdSchemaTask)
+default_registry.register("DWD_LOAD_FROM_ODS", DwdLoadTask)
+default_registry.register("DWD_QUALITY_CHECK", DwdQualityTask)
+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("DWS_BUILD_ORDER_SUMMARY", DwsBuildOrderSummaryTask)
 for code, task_cls in ODS_TASK_CLASSES.items():
     default_registry.register(code, task_cls)