python装饰器

python装饰器

装饰器（无参）

它是一个函数
函数作为它的形参。无参装饰器实际上就是一个单行参函数
返回值也是一个函数
可以使用@functionname方式，简化调用
注: 此处装饰器的定义只是就目前所学的总结，并不准确，只是方便理解
装饰器可以是高阶函数，但装饰器是对传入函数的功能的装饰（功能增强）
一个加法函数，想增加它的功能，能够输出被调用过以及调用的参数信息

def add(x,y):
    return x + y

def add(x,y):
    print("call add,x + y")
    return x + y

记录日志函数：

def add(x,y):
    print('add function:{} {}'.format(x,y))  #   日志记录
    return x + y

def add(x,y):
    return x + y

def logger(fn):
    print('before')
    print('add function:{} {}.format(4, 5)')
    ret = fn(4, 5)
    print('after',ret)
    return ret

logger(add)

before
add function:{} {}.format(4, 5)
after 9

9

参数解构的方法:

def add(x,y):
    return x + y

def logger(fn, *args, **kwargs):
    print('before')
    print('add function:{} | {}'.format(args, kwargs))
    ret = fn(*args, **kwargs)
    print('after', ret)
    return ret

logger(add, x=4, y=5)

before
add function:() | {'x': 4, 'y': 5}
after 9

9

另一种调用方法：

logger(add, 4, y=5)

before
add function:(4,) | {'y': 5}
after 9

9

上面的处理方法虽然是完成了需求，但是有以下的缺点：
1 打印是一个功能，这条语句和add函数耦合太高
2 加法函数属于业务功能，而输出信息的功能，属于非业务功能代码，不该放在业务函数add中
因此装饰器的作用就是在不侵入目标代码的情况下对目标代码就行一些增强作用

import datetime
import time
from functools import wraps

def logger(fn):
    @wraps(fn)
    def wrapper(*args, **kwargs):
        print("args = {}, kwargs = {}".format(args, kwargs))
        start = datetime.datetime.now() 
        
        ret = fn(*args, **kwargs)
        duration = datetime.datetime.now() - start
        print("function {} took {}s".format(fn.__name__, duration.total_seconds()))
        return ret
    return wrapper

@logger

def add(x,y):
    time.sleep(2)
    return x + y

print(add(3,y=4))

@logger是装饰器的一种语法
装饰器的运用类似于原函数就像是一幅画，装饰器是一个画框，有前置增强和后置增强功能，比如在画的前面加上一层玻璃保护画，在画的后面加上一层厚的木质保护层，共同完成对一副原始的画的装裱，但是并没有破坏原来的画的内容，仅仅是对这幅画起了装饰的作用。