37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
# -*- 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)
|
|
# 可以继续注册其他任务...
|