python 函数装饰器 优化代码结构

X1A0RAN / 2024-01-19 / 原文

# -*- coding: utf-8 -*-
# @Time    : 2023/11/9 9:18
# @Desc    : python 函数装饰器 优化代码结构


Func_Map = {}


# 注册装饰器
def register(language):
    def decorator(cls):
        Func_Map[language] = cls
        return cls

    return decorator


@register('a')
def func1():
    print('A')


@register('b')
def func2():
    print('B')


@register('')
def func3():
    print('C')


if __name__ == '__main__':
    for key, func in Func_Map.items():
        func()