一、装饰器是什么？

给原来的函数添加额外功能，但不需要修改原来函数的代码以及调用方式

二、装饰器的原则：

1.不修改原来函数
2.不修改原来函数的调用方式

三、装饰器的实现

1.不带参数的装饰器:

def decorat(func):
    print("xxxx")
    def wrapper():
        print("comein wrapper")
        start = time.time()
        func()
        end = time.time()
        print("fun1函数执行了{}s".format(end-start))
    return wrapper

@decorat
def fun1():
    time.sleep(3)
    print("func1...")
    return "hh"

fun1()

2.装饰器本身带参数:

def decorat2(unit):
    print("unit")
    def outter(func):
        def wrapper():
            print("comein wrapper")
            start = time.time()
            func()
            end = time.time()
            print("fun1函数执行了{}{}".format(end-start,unit))
        return wrapper
    return outter

@decorat2(unit='s')
def fun3():
    time.sleep(3)
    print("func2...")
    return "hh"

fun3()

3.被装饰的函数带参数

def decorat1(unit):
    print("unit")
    def outter(func):
        def wrapper(*args, **kwargs):
            print("comein wrapper")
            start = time.time()
            res = func(*args, **kwargs)
            print("res:",res)
            end = time.time()
            print("fun1函数执行了{}{}".format(end-start,unit))
            return res
        return wrapper
    return outter

@decorat1(unit='s')
def fun2(name):
    time.sleep(3)
    print("func2...",name)
    return "hh"

res = fun2('hello world')
print(res)

def wrapper(func):
    def inner(*args):
        print("我要开始计算了")
        res = func(*args)
        print("计算玩了哈")
        return res
    return inner



f = wrapper(add)
f(1,2)

image.png

def wrapper(func):
    def inner(*args):
        print("我要开始计算了")
        res = func(*args)
        print("计算玩了哈")
        return res
    return inner

@wrapper
def add(x,y):
    print("{}+{}={}".format(x,y,x+y))



add(2,3)

image.png

四、写装饰器的方法

1.首先定义一个装饰器函数，并且接受一个函数名作为形参（该函数就是被装饰函数，通过语法糖可自动实现赋值，我们只需要定义一个函数名形参即可）
2.写一个wrapper函数，嵌套函数，这个嵌套函数负责两件事，一是将新增功能的代码写在里面，二是通过外层装饰器中传入的函数名调用被装饰函数。
3.在外层装饰器函数中返回wrapper函数的地址。
4.如果被装饰函数带参数，则需要在调用被装饰函数的wrapper函数中加上类似*args, **kwargs的形参并在调用被装饰函数时传给被装饰函数。
5.如果装饰函数本身带参数，则需要在wrapper函数外面再增加一个outer_wrapper，并将被装饰函数的函数名传给outer_wrapper。装饰函数本身的参数则传给最外层的函数名。
6.如果被装饰函数带返回值，则在内部调用被装饰函数的wrapper函数中也要把返回值return回来。