装饰器
7.1 什么是装饰器
通过一个需求，一步一步来了解 Python 装饰器。首先有这么一个输出员工打卡信息的函数：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
def punch():
print('昵称：两点水 部门：做鸭事业部 上班打卡成功')
punch()
复制代码输出的结果：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
昵称：两点水 部门：做鸭事业部 上班打卡成功
复制代码然后，产品反馈，不行啊，怎么上班打卡没有具体的日期，加上打卡的具体日期吧，这应该很简单，分分钟解决啦。好吧，那就直接添加打印日期的代码吧，如下：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
import time
def punch():
print(time.strftime('%Y-%m-%d', time.localtime(time.time())))
print('昵称：小明 部门：研发部 上班打卡成功')
punch()
复制代码输出的结果：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
2018-01-09
昵称：小明 部门：研发部 上班打卡成功
复制代码这样改是可以，可是这样改是改变了函数的功能结构的，本身这个函数定义的时候就是打印某个员工的信息和提示打卡成功，现在增加打印日期的代码，可能会造成很多代码重复的问题。比如，还有一个地方只需要打印员工信息和打卡成功就行了，不需要日期，那么你又要重写一个函数吗？而且打印当前日期的这个功能方法是经常使用的，是可以作为公共函数给各个模块方法调用的。当然，这都是作为一个整体项目来考虑的。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
既然是这样，我们可以使用函数式编程来修改这部分的代码。因为通过之前的学习，我们知道 Python 函数有两个特点，函数也是一个对象，而且函数里可以嵌套函数，那么修改一下代码变成下面这个样子：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
import time
def punch():
print('昵称：小明 部门：研发部 上班打卡成功')
def add_time(func):
print(time.strftime('%Y-%m-%d', time.localtime(time.time())))
func()
add_time(punch)
复制代码输出的结果：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
2018-01-09
昵称：小明 部门：研发部 上班打卡成功
复制代码这样是不是发现，这样子就没有改动 punch 方法，而且任何需要用到打印当前日期的函数都可以把函数传进 add_time 就可以了。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
使用函数编程是不是很方便，但是，我们每次调用的时候，我们都不得不把原来的函数作为参数传递进去，还能不能有更好的实现方式呢？有的，就是本文要介绍的装饰器，因为装饰器的写法其实跟闭包是差不多的，不过没有了自由变量，那么这里直接给出上面那段代码的装饰器写法，来对比一下，装饰器的写法和函数式编程有啥不同。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
import time
def decorator(func):
def punch():
print(time.strftime('%Y-%m-%d', time.localtime(time.time())))
func()
return punch
def punch():
print('昵称：小明 部门：研发部 上班打卡成功')
f = decorator(punch)
f()
复制代码输出的结果：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
2018-01-09
昵称：两点水 部门：研发部 上班打卡成功
复制代码通过代码，能知道装饰器函数一般做这三件事：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
- 接收一个函数作为参数
- 嵌套一个包装函数, 包装函数会接收原函数的相同参数，并执行原函数，且还会执行附加功能
- 返回嵌套函数
7.2 语法糖
我们看上面的代码可以知道， Python 在引入装饰器 （Decorator） 的时候，没有引入任何新的语法特性，都是基于函数的语法特性。这也就说明了装饰器不是 Python 特有的，而是每个语言通用的一种编程思想。只不过 Python 设计出了 @ 语法糖，让定义装饰器，把装饰器调用原函数再把结果赋值为原函数的对象名的过程变得更加简单，方便，易操作，所以 Python 装饰器的核心可以说就是它的语法糖。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
那么怎么使用它的语法糖呢？很简单，根据上面的写法写完装饰器函数后，直接在原来的函数上加 @ 和装饰器的函数名。如下：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
import time
def decorator(func):
def punch():
print(time.strftime('%Y-%m-%d', time.localtime(time.time())))
func()
return punch
@decorator
def punch():
print('昵称：小明 部门：研发部 上班打卡成功')
punch()
复制代码输出结果：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
2018-01-09
昵称：小明 部门：研发部 上班打卡成功
复制代码那么这就很方便了，方便在我们的调用上，比如例子中的，使用了装饰器后，直接在原本的函数上加上装饰器的语法糖就可以了，本函数也无虚任何改变，调用的地方也不需修改。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
不过这里一直有个问题，就是输出打卡信息的是固定的，那么我们需要通过参数来传递，装饰器该怎么写呢？装饰器中的函数可以使用 *args 可变参数，可是仅仅使用 *args 是不能完全包括所有参数的情况，比如关键字参数就不能了，为了能兼容关键字参数，我们还需要加上 **kwargs 。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
因此，装饰器的最终形式可以写成这样：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
import time
def decorator(func):
def punch(*args, **kwargs):
print(time.strftime('%Y-%m-%d', time.localtime(time.time())))
func(*args, **kwargs)
return punch
@decorator
def punch(name, department):
print('昵称：{0} 部门：{1} 上班打卡成功'.format(name, department))
@decorator
def print_args(reason, **kwargs):
print(reason)
print(kwargs)
punch('小明', '研发部')
print_args('小明', sex='男', age=99)
复制代码输出的结果：文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/5161.html
2018-01-09
昵称：小明 部门：研发部 上班打卡成功
2018-01-09
小明
{'sex': '男', 'age': 99}