python装饰器实现函数注册和类装饰

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_39687901/article/details/82841825

一、使用装饰器实现回调函数注册

class FunctionManager:
    def __init__(self):
        print("初始化")
        self.functions = []

    def execute_all(self):
        for func in self.functions:
            func()

    def register(self, func):
        self.functions.append(func)


fm = FunctionManager()


@fm.register
def t1():
    print("t1")


@fm.register
def t2():
    print("t2")


@fm.register
def t3():
    print("t3")

fm.execute_all()

运行结果：

初始化
t1
t2
t3

二、使用装饰器获取类的信息

def class_info():
    def wrapper(cls):
        print(cls.__dict__)

    return wrapper


@class_info()
class Test:
    a = 1
    b = "123"
    c = ["66", "666"]
    d = {"name": "foxyball"}

运行结果：

{'__module__': '__main__', 'a': 1, 'b': '123', 'c': ['66', '666'], 'd': {'name': 'foxyball'}, '__dict__': <attribute '__dict__' of 'Test' objects>, '__weakref__': <attribute '__weakref__' of 'Test' objects>, '__doc__': None}