day05-装饰器作业

装饰器的重要内容
    functools.wraps
    带参数的装饰器
    多个装饰器装饰同一个函数



一.编写一个装饰器，为多个函数添加登陆认证的功能。具体：只需要登录一次,后面被装饰的函数不需要重新登陆。
FLAG = False
def wrapper(f):
    def inner(*args,**kwargs):
        global FLAG
        if FLAG == True:
            ret = f(*args, **kwargs)
            return ret
        else:
            username = input('username:')
            password = input('password:')
            if username == 'daxiong' and password == '136320':
                FLAG = True
                ret = f(*args,**kwargs)
                return ret
            else:
                print('登陆失败')
    return inner
@wrapper
def a():
    return '今晚中大奖'

@wrapper
def b():
    return '奖金八百万'

print(a())
print(b())


二.带参数的装饰器：
def wrapper(f):
    def inner(*args,**kwargs):
        ret = f(*args,**kwargs)
        return ret
    return inner
@wrapper
def func():
    return '双色球中奖800万'           #.__name__是求出字符串格式的函数名，用于文件操作。
print(func.__name__)                   #因为func=inner,执行func相当于执行inner,所以执行结果是inner

from functools import wraps           #加上这句话以及 @wraps(f)，print(func.__name__)的执行结果是func.否则就是inner。
def wrapper(f):
    @wraps(f)                          #带参数的装饰器，带参数的语法糖。
    def inner(*args,**kwargs):
        ret = f(*args,**kwargs)
        return ret
    return inner
@wrapper
def func():
    return '双色球中奖800万'
print(func.__name__)                   #执行结果是func


三.带参数的装饰器，装饰多个函数，方便好用。FLAG相当于开关，当开关是True,被装饰的函数走的是计算时间的通道，当开关是False时，走另外一条通道。
FLAG = True          #使用全局变量，True时，所有函数都被装饰，否则不被装饰。
import time
def wrapper(fl):        #让语法糖带参数(传入全局变量)，必须定义一个外层函数。最多三层。三层嵌套函数。
    def timmer(f):
        def inner(*args,**kwargs):
            global FLAG
            if FLAG:
                start = time.time()
                time.sleep(0.01)
                ret = f(*args,**kwargs)
                end = time.time()
                print(end - start)
                return ret
            else:
                print('当FLAG为False时，时间装饰器不起作用')
        return inner
    return timmer                  #timmer=wrapper(FLAG)
@wrapper(FLAG)                   #timmer=wrapper(FLAG)，所以@wrapper(FLAG)就是@timmer
def a():
    return '我要中双色球一等奖'
@wrapper(FLAG)
def b():
    return '奖金800万'
print(a())
print(b())

#注意：带参数的装饰器是三层函数，由参数 FLAG=1,2,其他 来控制选择哪一种功能。

FLAG = 1 #FLAG=1,2,其他
import time

def wapper(f1):
    def timmer(f):
        def inner(*args,**kwargs):
            global FLAG
            if FLAG == 1:
                start = time.time()
                time.sleep(0.1)
                n = f(*args, **kwargs)
                end = time.time()
                print(end-start)
                return n
            elif FLAG == 2:
                print('2222')
                n = f(*args, **kwargs)
                return n
            else:
                print('其他')
                n = f(*args, **kwargs)
                return n
        return inner
    return timmer

@wapper(FLAG)
def f1():
    print('f1111')

@wapper(FLAG)
def f2():
    print('f22222')

f1()
f2()

四.使用多个装饰器来装饰同一个函数。用处：记录用户的登录情况，计算这个函数的执行时间。
def wrapper1(f):
    def inner():
        print('wrapper1 before')
        f()
        print('wrapper1 after')
    return inner
def wrapper2(f):
    def inner():
        print('wrapper2 before')
        f()
        print('wrapper2 after')
    return inner

@wrapper2
@wrapper1
def func():
    print('明天继续加油')
func()
#wrapper2 before
#wrapper1 before
#明天继续加油
#wrapper1 after
#wrapper2 after