TASK_REGISTRY = {}

def register_task(name):
    """任务注册装饰器：@register_task('task_name')"""
    def decorator(cls):
        TASK_REGISTRY[name] = cls
        return cls
    return decorator

def get_task(name):
    """根据任务名获取 Task 类"""
    return TASK_REGISTRY.get(name)