装饰器

  • 定义：就是给已有函数增加额外功能的函数，它本质上就是一个闭包函数。
# 装饰器基本语法格式
def 装饰器函数(原函数fn):  # 原函数:被装饰的目标函数.
    def inner():
        '''执行函数之前'''
        fn()  // 执行被装饰的目标函数
        '''执行函数之后'''
    return inner
@装饰器函数
def f1():
    pass
    
# 灵魂语法
f1 = 装饰器函数(f1) 等价于 函数上写@装饰器函数
  • 作用：实现代码的重用
  • 特点：
  1. 不修改已有函数的源代码；
  2. 不修改已有函数的调用方式；
  3. 给已有函数增加额外的功能。

语法糖

  • 如果又多个函数都需要添加某种功能，每次都需要func = out_func(func)这样的代码对已有函数进行装饰很麻烦。
  • Python提供了一个装饰函数更加简单的写法，那就是语法糖，语法糖的书写格式是：@装饰器名字，使用“@”语法糖精简装饰器的代码的同时也可以完成对已有函数的装饰。
  • 说明：
  1. @check等价于comment = check(comment)
  2. 装饰器的执行时间是加载模块时立即执行。

通用装饰器

  • 需求：上面的普通装饰器，原函数既及不带参数也不带返回值，但是怎么设计可以不管函数有没有参数、返回值都可以兼容？
def 装饰器函数(原函数引用fn):
    def inner(*args, **kwargs):
        // 执行前扩展功能
        result = fn(*args, **kwargs)  # 将函数返回值暂存，合适的时机调用
        // 执行后扩展功能
        return result
    return inner
@装饰器函数
def fn(arg1, arg2,...):
    pass
    return xxxx