第7章：函数装饰器和闭包-实现简单装饰器和参数化装饰器

实现简单装饰器：

import time


def timer(func):  # 1. 装饰器接收被装饰的函数

    def wrapper(*args, **kwargs):  # 2. wrapper 接收被装饰函数的参数，然后将参数传给被装饰的函数：
        print(f'wrapper() 接收到位置参数{args}, 关键字参数{kwargs}')

        start_ti = time.time()
        result = func(*args, **kwargs)  # 3. 接收 wrapper 传来的参数并调用被装饰的函数
        total_ti = time.time() - start_ti
        print(f'函数执行耗时 {total_ti} 秒')
        return result

    return wrapper


@timer
def demo(n, s=1):
    print(f'Calling demo({n}, {s}) ，现在开始休眠 {n + s} 秒！')
    time.sleep(s + n)
    return 'X'


if __name__ == '__main__':
    x = demo(2, s=3)
    print('执行结果：', x)

wrapper() 接收到位置参数(2,), 关键字参数{'s': 3}
Calling demo(2, 3) ! 现在开始休眠 5 秒！
函数执行耗时 5.004848957061768 秒
执行结果：X

参数化装饰器：

装饰器也可以有参数，以下面的代码为例，但是从概念上看 timer 已经不是一个装饰器，而是一个装饰器工厂函数，timer 返回的 decorator 才是真正的装饰器；

import time


def timer(sleep_ti_1, sleep_ti_2=1):  # 1. 接收装饰器的参数
    print(f'timer() 接收到位置参数：{sleep_ti_1} ，关键字参数：{sleep_ti_2}！')

    # decorator 才是真正的装饰器
    def decorator(func):  # 2. decorator 接收被装饰的函数
        print(f'decorator() 接收到被装饰的函数：{func}!')
        print(f'decorator 层仍然可以使用 timer 的参数：{sleep_ti_1}, {sleep_ti_2}!')

        def wrapper(*args, **kwargs):  # 3. wrapper 接收的是被装饰函数的参数
            print(f'wrapper() 接收到被装饰的函数的位置参数：{args}，关键字参数：{kwargs}!')

            # 对 wrapper 来说，只有 func 才是自由变量，但是可以用 nonlocal 将 sleep_ti_x 声明为自由变量
            nonlocal sleep_ti_1, sleep_ti_2
            print(f'将 timer 的参数声明为自由变量：{sleep_ti_1}, {sleep_ti_2}!')

            start_ti = time.time()
            result = func(*args, **kwargs)  # 4. 接收 wrapper 传来的参数并调用被装饰的函数，
            end_ti = time.time()
            total_ti = end_ti - start_ti
            print(f'函数执行耗时 {total_ti} 秒')
            return result

        return wrapper

    return decorator


@timer(1, 2)
def demo(n, *, s=1):
    print(f'Calling demo({n}, {s})，现在开始休眠 {n + s} 秒！')
    time.sleep(n + s)
    return 'X'


if __name__ == '__main__':
    print('********** Calling __main__ Now **********')
    x = demo(2, s=3)
    print('执行结果：', x)

    # 查看 demo 的自由变量
    print(demo.__code__.co_freevars)

timer() 接收到位置参数：1 ，关键字参数：2！
decorator() 接收到被装饰的函数：<function demo at 0x10f3c0378>!
decorator 层仍然可以使用 timer 的参数：1, 2!
********** Calling __main__ Now **********
wrapper() 接收到被装饰的函数的位置参数：(2,)，关键字参数：{'s': 3}!
将 timer 的参数声明为自由变量：1, 2!
Calling demo(2, 3)，现在开始休眠 5 秒！
函数执行耗时 5.004558086395264 秒
执行结果： X
('func', 'sleep_ti_1', 'sleep_ti_2')

Process finished with exit code 0