register.py

import os

task_registry = {}
group_registry = {}
task2func_index = {}
func2task_index = {}

def register_task(name):
    def wrapper(func):

        task_registry[name] = func
        func2task_index[func.__name__] = name
        task2func_index[name] = func.__name__
        return func

    return wrapper

def register_group(name):
    def wrapper(func):

        func_name = func2task_index[func.__name__]

        if name in group_registry:
            group_registry[name].append(
                func_name
                )
        else:
            group_registry[name] = [func_name]
        return func
    return wrapper


# @register_group('group_a')
# @register_task('a')
# def foo():
#     pass

# @register_group('group_a')
# @register_task('b')
# def fii():
#     pass

# @register_group('group_b')
# @register_task('c')
# def bar():
#     pass

# name = 'A'  # or args.type
# func_to_call = REGISTER[name]
# func_to_call()  # actual call is done here