Python笔记4（装饰器进阶）

进阶的需求

1、第一种情况

需要给500个函数添加装饰器，如果你逐个添加或者取消添加会非常麻烦，并且浪费成本；
现在，你可以设计你的装饰器 来确认是否生效。

import time
FLAG = True#全局变量
def outer(flag):
    def timmer(f):
        def inner(*args,**kwargs):
            if flag == True:#如果flag为True，原有的函数前后增加功能，且不改变原函数的调用方式
                start_time = time.time()
                ret = f(*args,**kwargs)
                end_time = time.time()
                print(end_time - start_time)
            else:#如果flag不为True，只执行原有的函数，且不改变原函数的调用方式
                ret = f(*args, **kwargs)
            return ret
        return inner
    return timmer

@outer(FLAG)   # func = timmer(func)
def func(a,b):
    print('begin func',a)
    time.sleep(0.1)
    print('end func',b)
    return True

func(1,2)

#执行结果
#>>>begin func 1
#>>>end func 2
#>>>0.10027241706848145

2、第二种情况

装饰器除了可以计算函数的执行时间，还可以记录登录、记录日志等；
现在假如要访问博客园首页或其他内容的前提，需要用户先登录才能访问，设计装饰器。

import time
login_info = {'alex':False}
def login(func):   # manager
    def inner(name):
        if login_info[name] != True:
            user = input('user ：')
            pwd = input('pwd ：')
            if user == 'alex' and pwd == 'alex3714':
                login_info[name] = True
        if login_info[name] == True:
            ret = func(name)     # timmer中的inner
            return ret
    return inner

@login
def index(name):
    print('欢迎%s来到博客园首页~'%name)

@login
def manager(name):
    print('欢迎%s来到博客园管理页~'%name)

index('alex')#由于login_info[('alex']==Flase，所以第一次访问需要登录
index('alex')
manager('alex')
manager('alex')

#执行结果
#>>>user ：alex
#>>>pwd ：alex3714
#>>>欢迎alex来到博客园首页~
#>>>欢迎alex来到博客园首页~
#>>>欢迎alex来到博客园管理页~
#>>>欢迎alex来到博客园管理页~

3、第三种情况

1）因为要用到两个装饰器，开始之前先解释一下两个装饰器装饰一个函数的执行过程：

def wrapper1(func): 
    def inner1():
        print('wrapper1 ,before func')
        func() 
        print('wrapper1 ,after func')
    return inner1

def wrapper2(func): 
    def inner2():
        print('wrapper2 ,before func')
        func() 
        print('wrapper2 ,after func')
    return inner2

@wrapper2 
@wrapper1 
def f():
    print('in f')

f()

2）在第二种情况的基础上还要计算函数index、manger的执行时间：

import time
login_info = {'alex':False}
def login(func):   # manager
    def inner(name):
        if login_info[name] != True:
            user = input('user ：')
            pwd = input('pwd ：')
            if user == 'alex' and pwd == 'alex3714':
                login_info[name] = True
        if login_info[name] == True:
            ret = func(name)     # timmer中的inner
            return ret
    return inner

def timmer(f):
    def inner(*args,**kwargs):
        start_time = time.time()
        ret = f(*args,**kwargs)     # 调用被装饰的方法
        end_time = time.time()      #
        print(end_time - start_time)
        return ret
    return inner

@login
@timmer
def index(name):
    print('欢迎%s来到博客园首页~'%name)

@login
@timmer    # manager = login(manager)
def manager(name):
    print('欢迎%s来到博客园管理页~'%name)

index('alex')
index('alex')
manager('alex')
manager('alex')