File size: 302 Bytes
d8bfe4a
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
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)