迁移代码到Git

etl_billiards/orchestration/task_registry.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+"""任务注册表"""
+from tasks.orders_task import OrdersTask
+from tasks.payments_task import PaymentsTask
+from tasks.members_task import MembersTask
+
+class TaskRegistry:
+    """任务注册和工厂"""
+    
+    def __init__(self):
+        self._tasks = {}
+    
+    def register(self, task_code: str, task_class):
+        """注册任务类"""
+        self._tasks[task_code.upper()] = task_class
+    
+    def create_task(self, task_code: str, config, db_connection, api_client, logger):
+        """创建任务实例"""
+        task_code = task_code.upper()
+        if task_code not in self._tasks:
+            raise ValueError(f"未知的任务类型: {task_code}")
+        
+        task_class = self._tasks[task_code]
+        return task_class(config, db_connection, api_client, logger)
+    
+    def get_all_task_codes(self) -> list:
+        """获取所有已注册的任务代码"""
+        return list(self._tasks.keys())
+
+
+# 默认注册表
+default_registry = TaskRegistry()
+default_registry.register("ORDERS", OrdersTask)
+default_registry.register("PAYMENTS", PaymentsTask)
+default_registry.register("MEMBERS", MembersTask)
+# 可以继续注册其他任务...