Python装饰器实用教程

最新推荐文章于 2024-08-06 15:14:11 发布

GondorFu

最新推荐文章于 2024-08-06 15:14:11 发布

阅读量148

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/a40850273/article/details/124930085

版权

本文介绍了Python中最简单的装饰器示例，包括普通函数装饰器和类装饰器的使用。通过代码示例展示了如何在函数前后添加额外逻辑，并解释了类函数调用时self和cls的区别以及classmethod的作用。同时，强调了类装饰器中classmethod与普通装饰器的调用顺序重要性。

摘要由CSDN通过智能技术生成

最简单的装饰器示例

def print_decripter(func):
    def wrapper(*args, **kwargs):
        print('wrapper start')
        tmp = func(*args, **kwargs)
        print(f'wrapper sum: {tmp}')
        return tmp
    return wrapper


@print_decripter
def add(a, b):
    return a + b


def main():
    print('add sum: {}'.format(add(1, 2)))


if __name__ == '__main__':
    main()

'''
wrapper start
wrapper sum: 3
add sum: 3
'''

类内装饰器示例

class Tmp:
    const_added_num = 1

    def print_decripter(func):
        def wrapper(*args, **kwargs):
            print('wrapper start')
            tmp = func(*args, **kwargs)
            print(f'sum: {tmp}')
            return tmp
        return wrapper

    @classmethod  # 装饰器顺序不能调整，不然报错
    @print_decripter
    def add(cls, a, b):
        return a + b + cls.const_added_num


def main():
    print('add sum: {}'.format(Tmp.add(1, 2)))


if __name__ == '__main__':
    main()

'''
wrapper start
sum: 4
add sum: 4
'''

注意事项：

类函数调用：
1. 当使用类实例进行调用时（Tmp().add()），函数内部会自动在调用参数的最前面添加 self，也就是该类实例。
2. 而当直接使用类名调用时（Tmp.add()），函数内部不会进行任何操作，直接将原始输入参数参入函数中。如果函数内部需要使用类对象，那么在函数定义时需要通过 @classmethod 声明为类函数。此时，函数内部的一个操作是在调用参数最前面添加 cls，也就是类对象。因此 classmethod 可以理解为 Python 内部实现的一个装饰器。
3. 不过 classmethod 并不会返回一个有效的函数，因此上述装饰器的调用顺序必须是最后调用 classmethod。此时的逻辑是：
  1. 首先 print_decripter 装饰器将 add 函数的前后分别添加相关输出逻辑，此时函数输入只有数字 1 和 2。
  2. 之后返回封装好的函数 wrapper，并将 wrapper 输入给 classmethod。
  3. 之后 classmethod 在函数参数最前面插入 cls，使其能够使用类对象。