Python3装饰器详解

装饰器

​ 装饰器本质上是一个函数，该函数用来处理其他函数，它可以让其他函数在不需要修改代码的前提下增加额外的功能。装饰器的返回值也是一个函数对象，它经常用于有切面需求的场景，比如：插入日志、性能测试、事物处理、缓存、权限校验等应用场景。装饰器是解决这类问题的绝佳设计，有了装饰器，我们就可以抽离出大量于函数功能本身无关的雷同代码继续重用。概括的讲，装饰器的作用就是为已经存在的对象添加额外的功能。

​ 一个装饰器可以用来装饰多个函数，一个函数也可以被多个装饰器装饰 。

​ @符号是装饰器的语法糖，在定义函数的时候使用，避免再一次赋值操作。

无参装饰器

​ 它是一个函数，被装饰函数作为它的形参，返回值也是一个函数，使用@functionname方式调用。装饰器是高阶函数，装饰器是对传入函数的功能的装饰（增强），不侵入任何代码。

def logger(fn):
    def wrapper(*args, **kwargs):
        print("args={}, kwargs={}".format(args,kwargs))
        ret = fn(*args, **kwargs)
        print("{}".format(fn.__name__))
        return ret
    return wrapper

@logger             # add = logger(add),返回内层函数的引用
def add(x, y):
    return x + y

print(add(4,5))

带参装饰器

​ 带参装饰器是在装饰器外层又加了一层函数

def logger(*paras):
    def _logger(fn):
        def wrapper(*args, **kwargs):
            print("args={}, kwargs={}".format(args,kwargs))
            ret = fn(*args, **kwargs)
            print("paras={}".format(paras))
            print("{}".format(fn.__name__))
            return ret
        return wrapper
    return _logger

@logger(10,20,30)      # add = logger(add),返回内层函数的引用
def add(x, y):
    return x + y

print(add(4,5))

functools模块

​ 被装饰后的函数其实已经是另外一个函数了（函数名等函数属性会发生改变）。为了不影响，Python的functools包中提供了一个叫wraps的decorator来消除这样的副作用。写一个decorator的时候，最好在实现之前加上functools的wraps或者update_wrapper，它能保留原有函数的一些内置属性，，例如：__name__、 __module__ 、__doc__ 和__dict__等。

import functools

def logger(fn):
    # 方法一
    @functools.wraps(fn)
    def wrapper(*args, **kwargs):
        print("args={}, kwargs={}".format(args,kwargs))
        print(id(fn))
        ret = fn(*args, **kwargs)
        print("{}".format(fn.__name__))
        return ret
    # 方法二
    # functools.update_wrapper(wrapper, fn)
    return wrapper

@logger             # add = logger(add),返回内层函数的引用
def add(x, y):
    return x + y

print(add(4,5))
print(add.__name__)
print(id(add.__wrapped__))