装饰器的进阶_2

案例1

import time
def timer(func):
    def inner(*args,**kwargs):
        start=time.time()
        ret=func(*args,**kwargs)
        end=time.time()
        print(end-start)
        return ret
    return inner

@timer
def func1():
    time.sleep(0.1)
    print('123')
    
@timer
def func2():
    time.sleep(0.1)
    print('1234')

func1()
func2()

需求，对于func1,func2，加入不需要使用装饰器，那么直接注释，需要注释2次；如果是500个那么就需要注释500次，这样非常麻烦。有什么办法，在装饰器外面再套上一层函数

import time
FLAG=False#也就是说当有500多个函数需要装饰器的时候，那么只需要控制这里的true和false据可以完成对全局的控制
#解决方案就是在装饰器前面再套上一层函数，对于装饰器的嵌套最多只能嵌套三层
def timer_out(flag):
    def timer(func):
        def inner(*args,**kwargs):
            if flag:#这句代码就相当于if flag=true,只不过true可以省略
                start=time.time()
                ret=func(*args,**kwargs)
                end=time.time()
                print(end-start)
                return ret
            else:#注意else和if条件后面唯一不同的就是缺少start和end语句，除此之外其他语句均相同
                ret = func(*args, **kwargs)
                return ret
        return inner
    return timer#注意这里是timer,而不是timerout

# # timer=timer_out(FLAG)
# @timer 等价于func1=timer(func1)
@timer_out(FLAG)#注意这里是和以前装饰器最大的不同，括号内有参数.其次，可以这样理解，首先timer_out()表示执行timer_out函数，然后@这个执行的结果；相当于func1=timer(func1)
# 等价于下面两行代码
# timer=timer_out(FLAG)
# @timer 等价于func1=timer(func1)
def func1():
    time.sleep(0.1)
    print('123')

@timer_out(FLAG)
def func2():
    time.sleep(0.1)
    print('1234')

func1()
func2()

代码的解释：

 双层装饰器

#多个装饰器装饰一个函数
def wrapper1(func):#这里将f传进来
    def inner1():
        print('wrapper1,before func')
        func()
        print('wrapper1,after func')
    return inner1

def wrapper2(func):#注意此时这个func指代的是inner1
    def inner2():
        print('wrapper2,before func')
        func()
        print('wrapper2,after func')
    return inner2

@wrapper1 #紧接着f=wrapper1(f)=wrapper1(inner1)
@wrapper2 #f=wrapper2(f)=inner2，注意装饰器是先找离他最近的函数进行执行，因此首先执行的是wrapper2
def f():
    print('in f')

f()#本质上调用的是inner2

 

上述代码可以用于场景:

1、记录用户的登录情况

2、记录记录用户的登录时间