浅析Python装饰器

一、什么是装饰器

在介绍装饰器以前，咱们先来思考一个问题：使用Python语言进行程序设计时，若是咱们想扩展一个函数的功能，通常会怎么作呢？

好比，有一个名为print_info函数，当前该函数内只作一些简单的打印操做，如今咱们想扩展这个函数功能，如在发生错误时，咱们将错误行号传入到该函数打印出来。

def print_info():
    print("Hello World")

看到这个问题，咱们的第一反应确定是从新修改这个函数。在print_info函数增长一个参数，用于接收外部传入的错误行号，在函数内部增长打印行号信息语句。

def print_info(err_line):
    print("Hello World")
    print("The Error Occurred Line Number: %d" %(err_line))

彷佛，经过修改函数内容，也能实现咱们的需求。但有没有这种可能呢，不直接进行修改print_info函数内容，经过其余的方式增长print_info函数的额外功能。答案确定是有的，能实现这种功能的就是咱们要讲的装饰器。

装饰器究竟是什么呢？使用比较严谨的语言来描述：

装饰器本质上是一个Python函数，它可让其余函数在不须要作任何代码变更的前提下增长额外功能，装饰器的返回值也是一个函数对象。它常常用于有切面需求的场景，好比：插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计，有了装饰器，
咱们就能够抽离出大量与函数功能自己无关的雷同代码并继续重用。

 归纳的讲，装饰器的做用就是为已经存在的函数或对象添加额外的功能。

二、怎么写一个装饰器

咱们先来看一段这样的代码

 1 import sys
 2 
 3 def log(fcn):
 4     def wrapper(*argc, **kw):
 5         #New Operations Are Added Here
 6         if len(argc) != 1:
 7             print('Illegal parameter')
 8             return -1
 9         print("The Error Occurred Line Number: %d" %(argc[0]))
10         return fcn()
11     return wrapper      
12 
13 def print_info():
14     print("Hello World")
15 
16 print_info = log(print_info) 
17 
18 def main():
19     print_info(sys._getframe().f_lineno) # sys._getframe().f_lineno表明当前的行号
20 
21 if __name__ == "__main__":
22     main()# 添加功能并保持原函数名不变

log函数是一个“闭包”（关于什么是“闭包”可参考“浅析Python闭包”），该函数有一个参数fcn，返回值是内部实现的wrapper函数。

比较重要的一条语句print_info = log(print_info)，这里经过对print_info变量赋值改变它原来指向的类型，起到了对print_info函数赋予了新功能。执行这句语句以后print_info变量再也不表示前面定义的print_info函数，而是表示log函数内实现的wrapper函数，但原来定义的print_info函数仍然存在。

最终调用的print_info函数，其实调用的是log中的wrapper函数。因为“闭包”特性，虽然wrapper函数已经离开了创造它的环境log函数，但它仍然可使用log函数中的自由变量。wrapper函数中，打印了传入的错误行号，并调用了传入的原定义的print_info函数，这样便作到了对定义的print_info函数内容作修改，调用print_info增长了额外功能。

运行结果

The Error Occurred Line Number: 19 # 打印出新增的对应行号信息，在19行被调用，因此传入行号是19
Hello World #原print_info函数执行内容

上面的log函数其实已是一个装饰器了，它对原函数作了包装并返回了另一个函数，额外添加了一些功能。由于这样写实在不太优雅，新版本Python中支持了@语法糖，咱们能够把它使用起来。

下面代码等同于上面的写法。

import sys

def log(fcn):
    def wrapper(*argc, **kw):
        #New Operations Are Added Here
        if len(argc) != 1:
            print('Illegal parameter')
            return -1
        print("The Error Occurred Line Number: %d" %(argc[0]))
        return fcn()
    return wrapper    
    
@log                      
def print_info():
    print("Hello World")

#print_info = log(print_info)

def main():
    print_info(sys._getframe().f_lineno)

if __name__ == "__main__":
    main()

把@log放到print_info函数定义处，至关于执行了print_info = log(print_info)语句

三、完整装饰器的写法

前面写的装饰器都没有问题，可是还差最后一步，虽然装饰器装饰过的函数看上去名字没变，其实已经变了。

def log(fcn):
    def wrapper(*args, **kw):
        print(fcn.__name__)
        return fcn()
    return wrapper  

@log
def print_info():
    print('Hello World')
    
def main():
    print_info()
    print(print_info.__name__)
    
if __name__ == "__main__":
    main()

运行结果

print_info
Hello World
wrapper
>>> 

上面print_info函数经log装饰器装饰后，它的__name__属性已经从原来的‘print_info’变成了‘wrapper’

由于返回的那个wrapper()函数名字就是'wrapper'，因此，须要把原始函数的__name__等属性复制到wrapper()函数中，不然，有些依赖函数签名的代码执行就会出错。

不须要在内部添加wrapper.__name__=fcn.__name__这样的代码，Python内置的functools.wraps就是干这个事的，因此，一个完整的decorator的写法以下：

import functools

def log(fcn):
    @functools.wraps(fcn)
    def wrapper(*args, **kw):
        print(fcn.__name__)
        return fcn()
    return wrapper

四、高阶一点装饰器

4.1 带参数装饰器

若是装饰器函数自己要传入参数，那么装饰器就会是这样的

import sys

def log(text):
    def wrapper(fcn)
        def inner_wrapper(*argc, **kw):
            #New Operations Are Added Here
            print(text)
            if len(argc) != 1:
                print('Illegal parameter')
                return -1
            print("The Error Occurred Line Number: %d" %(argc[0]))
            return fcn()
        return inner_wrapper
    return wrapper    
    
@log('With Parameter Decorator')
def print_info():
    print("Hello World")

#print_info = log(print_info)

def main():
    print_info(sys._getframe().f_lineno)

if __name__ == "__main__":
    main()

对于带参数装饰器，能够这么理解，当带参数的装饰器被装饰在某个函数上时，好比上述代码@log('With Parameter Decorator')，log('With Parameter Decorator')实际上是一个函数，会立刻被执行，它返回的结果任是一个装饰器wrapper。

也就是下面代码其实等价于print_info = wrapper(print_info)

@log('With Parameter Decorator')
def print_info():
    print("Hello World")

4.2 装饰器装饰类方法

前面咱们介绍的都是用装饰器装饰函数，装饰器同时也能够装饰类方法，下面来看如何用装饰器装饰类的方法

咱们定义了名为simple_test的类，这个类有一个data属性和get_data方法。类里的实现很是简单，get_data方法只返回data属性值，并不进行其余操做。同时，咱们还需定义了一个装饰器装饰类的get_data方法，装饰器中根据数据获取源来编写其中功能，好比咱们这里选择从串口终端获取数据，直接修改装饰器的内容让其获取终端输入数据。

这样在设计一个类时，作到了将类中稳定的部分和常常要修改的部分独立开。类设计完成后，咱们不须要更改类的源码，在使用时只须要根据需求修改类方法的装饰器就能实现咱们的需求。

def serial_data(fcn):
    def wrapper(self, *args, **kw):
        str = input('Please enter an integer: ')
        try:
            tmp = int(str)
        except ValueError:
            print('Invalid Value')
            return None
        self.data = tmp
        return fcn(self)
    return wrapper

class simple_test(object):
    def __init__(self, _val = 0):
        self.data = _val
    
    @serial_data
    def get_data(self):
        return self.data
    
def main():
    obj = simple_test()
    val = obj.get_data()
    if val is not None:
        print('get Data: %d' %(val))
    
if __name__ == "__main__":
    main() 

运行结果：

Please enter an integer: 100
get Data: 100
>>>