python语言基础-5 进阶语法-5.2 装饰器-5.2.2 简单装饰器

声明：本内容非盈利性质，也不支持任何组织或个人将其用作盈利用途。本内容来源于参考书或网站，会尽量附上原文链接，并鼓励大家看原文。侵删。

5.2.2 简单装饰器

装饰器的形式就是一个闭包，下面是一个简单的定义并使用装饰器的例子。如下：

'''
def house():print('我是毛坯房……')
这是一个原函数，下面我们要自定义一个装饰器，并为这个原函数添加装饰器的效果。
'''# 定义一个装饰器
def decorate(func):  # 定义装饰器函数，它的参数是一个函数a = 100print('wrapper开始加载......')def wrapper():func()  # func()代表被装饰的原函数，使用中可根据需要放置其位置print('----刷漆')print('----铺地板', a)print('----做装潢')print('wrapper加载完成......')return wrapper  # 返回值是内部函数# 定义函数并使用装饰器
@decorate  # 装饰器在定义被装饰函数时使用，自定义装饰器的名称就是我们定义装饰器函数（闭包函数）时的函数名
def house():print('我是毛坯房……')# 调用函数，可以看出打印结果中带有装饰器的效果
house()# 打印函数名查看函数的特性
print(house)  # 输出结果为<function decorate.<locals>.wrapper at 0x0000029C7675D310>，打印house，但输出wrapper，是因为装饰器装返回值给了house

由上面的例子可以看出，装饰器有如下特点：

• 装饰器是一个闭包；
• 装饰器以函数为参数。

为什么在被装饰器装饰后，调用原函数会有装饰效果？我们仔细考虑一下装饰器的作用过程：

1. 装饰器的效果被打印了（上面的例子中），说明装饰器被调用了；
2. 打印函数名显示内存地址时出现了与装饰器内层函数相关的内容（上面例子中），说明函数的地址是指向装饰器内层函数的地址的；
3. 因此，我们可以推测：当调用原函数时，装饰器内层函数会被返回给了被装饰函数，即被装饰后装饰器的内层函数就代表了被装饰函数。

即然装饰器的内层函数代表了被装饰函数，那么需要我们考虑一种情况：当被装饰函数有参数时，装饰器的内层函数需要不需要保持参数列表的对应（不一定是一致）？答案是肯定的。如下：

'''
被装饰函数有参数，而装饰器的内层函数没有参数时，运行会报错
'''
def decorate(func):  # 定义装饰器函数a = 100print('wrapper开始加载......')def wrapper():  # 原函数有参，此处没有参数func(8)  # func(x)相当于原函数的调用，直接写定一个参数print('----刷漆')print('----铺地板', a)print('----做装潢')print('wrapper加载完成......')return wrapper# 使用装饰器
@decorate
def house(n):print('我是毛坯房{}'.format(n))house(3)  # 定义的原函数有参数，调用时参数是不能为空的# 运行结果会报如下错误：TypeError: decorate.<locals>.wrapper() takes 0 positional arguments but 1 was given，这说明我们一定要给装饰器的内层函数设置参数'''
被装饰函数有参数，装饰器的内层函数也需要参数
'''
def decorate(func):  # 定义装饰器函数a = 100print('wrapper开始加载......')def wrapper(*x):  # 原函数有参，因此此处也要有参，可以是单参数x，也可以是可变参数 *xfunc(x)  # func(x)相当于原函数的调用，也要有参，可以是可变参数 *xprint('----刷漆')print('----铺地板', a)print('----做装潢{}'.format(x))print('wrapper加载完成......')return wrapper# 使用装饰器
@decorate
def house(n):print('我是毛坯房{}'.format(n))house(3)  # 定义的原函数有参数，调用时参数是不能为空的