装饰器从入门到精通

装饰器从入门到精通:
1.背景：年底到了，又要发年终奖了，可是怎么发比较好呢？
　　你碰到了一个不懂技术的老大，老大说了，把大家把写得最为
　　得意的一段代码拿出来跑时间，咱们凭实力说话!!!于是乎老板
　　把测试时间的任务交给你了，那么问题来了，到底怎么解决好呢？

2.于是乎，经过仔细思考你想到了一个方案，以前学数据结构的时候不是
　　测试代码效率的办法：引用系统自带的时间模块即可:

　　import time

　　start = time.time()
　　func()
　　end = time.time()
　　print(end - start)

3.开放封闭原则:
　　开放：对扩展是开放的
　　封闭：对修改是封闭的
4.好了不废话了，很显然，上面的代码是不满足这个原则的，接下来
　　今天的主角登场了，装饰器！！！
5.初识装饰器:

import time


def timmer(f):
    def inner():
        start = time.time()
        f()
        end = time.time()
        print(end - start)

    return inner


def func():
    time.sleep(0.1)
    # 时间太短无法测试 所以sleep一下
    print("hello world")


func = timmer(func)
# 这里面讲func作为参数传到里面去执行了一趟!!!
# 返回的func实际上是inner
func()
"""
hello world
0.10023927688598633
"""

实际上这就是一个闭包嘛，地球人真麻烦取这么多名字干嘛!!!

6.语法糖
　　老是这么写：func = timmer(func)，实在是繁琐，于是呢人们又发明了一个
　　新的写法来代替上面这句话：在被装饰的函数前面加上@起装饰作用的函数
　　上面这个就写成这样了:

import time


def timmer(f):
    def inner():
        start = time.time()
        f()
        end = time.time()
        print(end - start)

    return inner


@timmer  # 相当于 func = timmer(func)
def func():
    time.sleep(0.1)
    # 时间太短无法测试 所以sleep一下
    print("hello world")


func()
"""
hello world
0.10023927688598633
"""

7.参数问题

　　如果被装饰的函数里面有参数怎么办？装饰之后的func实际上是inner，而func又在

inner里面执行，所以我把参数给inner，顺便再返回一下就可以了。

import time


def timmer(f):
    def inner(*args, **kwargs):
        start = time.time()
        ret = f(*args, **kwargs)
        end = time.time()
        print(end - start)
        return ret

    return inner


@timmer  # 相当于 func = timmer(func)
def func(a, b):
    time.sleep(0.1)
    # 时间太短无法测试 所以sleep一下
    print("hello world")
    return a + b


a = func(1, 2)
print(a)
"""
hello world
0.10096383094787598
3
"""

8.装饰器公式 ===>万能金刚

import time


def outer(f):
    def inner(*args, **kwargs):
        start = time.time()
        ret = f(*args, **kwargs)
        end = time.time()
        print(start - end)
        return ret

    return inner


@outer
def func():
    pass

9.高阶装饰器（带参数的装饰器）

def route(rule, **options):  # 从右往左执行 第一层实际上是传递参数
    def decorator(f):
        def inner(*args, **kwargs):
            print(options)
            # dic =
            print({**options})
            # print(**options)
            endpoint = options.pop("endpoint", f.__name__)  # 弹出字典里面的值
            print("被装饰的函数为:", endpoint)
            print(rule)
            res = f(*args, **kwargs)
            print("in decorator")
            print(id(f))
            return res

        return inner

    return decorator


dic = {"name": "dgf", "age": 18}


@route('/我是路由/', name="dgf")  # 这句话等同于 func=outer(func)
def printIn(str):
    # print(id(printIn))
    print('h')
    return str


ret = printIn("hello world")
print(ret)

import time


def timmer_out(flag):
    def timmer(f):
        def inner(*args, **kwargs):
            if flag:
                start = time.time()
                ret = f(*args, **kwargs)
                end = time.time()
                print(end - start)
                return ret
            else:
                ret = f(*args, **kwargs)
                return ret

        return inner

    return timmer


@timmer_out(True)
def func():
    time.sleep(0.1)
    print("我是func")


func()
"""
我是func
0.10094308853149414
"""

再将参数改为FALSE

import time


def timmer_out(flag):
    def timmer(f):
        def inner(*args, **kwargs):
            if flag:
                start = time.time()
                ret = f(*args, **kwargs)
                end = time.time()
                print(end - start)
                return ret
            else:
                ret = f(*args, **kwargs)
                return ret

        return inner

    return timmer


@timmer_out(False)
def func():
    time.sleep(0.1)
    print("我是func")


func()
"""
我是func
"""

这里实际上要分两步看这个问题

10.多个装饰器装饰一个函数

def func1(f):
    def inner(*args, **kwargs):
        print("fun1 start")
        func_ret = f()
        print("fun1 end")
        return func_ret

    return inner


def func2(f):
    def inner(*args, **kwargs):
        print("fun2 start")
        func_ret = f()
        print("fun2 end")
        return func_ret

    return inner


def func3(f):
    def inner(*args, **kwargs):
        print("fun3 start")
        func_ret = f()
        print("fun3 end")
        return func_ret

    return inner


@func3
@func2
@func1
def func():
    print("func")


if __name__ == "__main__":
    func()

"""
fun3 start
fun2 start
fun1 start
func
fun1 end
fun2 end
fun3 end

"""