12.Python（装饰器）

一、定义

装饰器：本质是函数，为其他函数添加附加功能

原则：

1、不修改被修饰函数的源代码

2、不修改被修饰函数的调用方式

 装饰器=高阶函数+函数嵌套+闭包

二、高阶函数

1.函数接收的参数是一个函数名

2.函数的返回值是一个函数名

3.满足上述其中一个条件就是高阶函数

下例中是一个高阶函数：

def foo():
    print("你好")

def test(func):#是一个高阶函数
    print(func)#func有了foo的内存地址
    func()#运行foo函数
test(foo)

def foo():
    print("来自foo")

def test(func):#接收参数是函数名
    func()
    return func#返回值还是函数名
foo=test(foo)
foo()#还是执行原来函数的调用方式

 三、装饰器实现

1、基本实现

import time

def timmer(func): #func=test
    def wrapper():
        start_time = time.time()
        func() #就是在运行test()
        stop_time = time.time()
        print('运行时间是%s' % (stop_time-start_time))
    return wrapper

@timmer #test=timmer(test)
def test():
    time.sleep(3)
    print('test函数运行完毕')


test()  #执行的是wrapper()

 2、原本的函数有返回值

import time

def  timmer(func):  #func=test
    def wrapper():
        start_time = time.time()
        res = func()   #  就是在运行test()
        stop_time = time.time()
        print('运行时间是%s' % (stop_time-start_time))
        return res
    return wrapper

@timmer  #test=timmer(test)
def test():
    time.sleep(3)
    print('test函数运行完毕')
    return "这是test的返回值"

res = test()  
print(res)

 3、加上参数

import time
def timmer(func): #func=test1
    def wrapper(*args, **kwargs): #固定用法，不管原函数有几个参数，这种形式都能接收
        start_time = time.time()
        res = func(*args, **kwargs) #就是在运行test1()  func(*('alex'),**{'age':18;'gender':male;....})
        stop_time = time.time()
        print('运行时间是%s' % (stop_time-start_time))
        return res
    return wrapper

@timmer
def test1(name, age, gender, habit):
    time.sleep(1)
    print('test1函数运行完毕,名字是【%s】 年龄是【%s】 ,性别是【%s】,爱好是【%s】' %(name, age, gender, habit))
    return '这是test的返回值'

test1('alex',18,'male',"玩")

4、验证功能

user_list=[
    {'name':'alex','passwd':'123'},
    {'name':'linhaifeng','passwd':'123'},
    {'name':'wupeiqi','passwd':'123'},
    {'name':'yuanhao','passwd':'123'},
]
current_dic={'username':None,'login':False} #全局变量，登录状态


def auth_func(func):
    def wrapper(*args, **kwargs):
        if current_dic['username'] and current_dic['login']:#有人开始登录,没人登录的话就直接不往下运行了
            return 0
        username = input('用户名：').strip()#用户输入用户名，程序把有输入内容前面的空格去掉
        passwd = input('密码：').strip()
        for user_dic in user_list:
            if username == user_dic['name'] and passwd == user_dic['passwd']:
                    current_dic['username']=username#改变登录状态，改变全局变量
                    current_dic['login']=True#意思就是现在有人登录
                    res = func(*args, **kwargs)
                    return res
        else:#for也可以加else，说明for循环中没有匹配的
                print('用户名或者密码错误')

    return wrapper

@auth_func
def index():
    print('欢迎来到京东主页')

@auth_func
def home(name):
    print('欢迎回家%s' %name)

@auth_func
def shopping_car(name):
    print('%s的购物车里有［%s,%s,%s］' %(name, '奶茶', '妹妹', '娃娃'))

print('before-->', current_dic)#先打印出登录状态
index()
print('after--->', current_dic)#输入登录内容后的登录状态
home('产品经理')

 5、带参数验证功能装饰器

user_list=[
    {'name':'alex','passwd':'123'},
    {'name':'linhaifeng','passwd':'123'},
    {'name':'wupeiqi','passwd':'123'},
    {'name':'yuanhao','passwd':'123'},
]
current_dic={'username':None,'login':False}

def auth(auth_type='filedb'):#在原来的装饰器基础上往上一层嵌套一个函数，变成一个新的装饰器
    def auth_func(func):
        def wrapper(*args,**kwargs):
            print('认证类型是',auth_type)
            if auth_type == 'filedb':
                if current_dic['username'] and current_dic['login']:
                    return 0
                username=input('用户名：').strip()
                passwd=input('密码：').strip()
                for user_dic in user_list:
                    if username == user_dic['name'] and passwd == user_dic['passwd']:
                        current_dic['username']=username
                        current_dic['login']=True
                        res = func(*args, **kwargs)
                        return res
                else:
                    print('用户名或者密码错误')
            elif auth_type == 'ldap':
                print('鬼才特么会玩')
                res = func(*args, **kwargs)
                return res
            else:
                print('鬼才知道你用的什么认证方式')
                res = func(*args, **kwargs)
                return res

        return wrapper
    return auth_func

@auth(auth_type='filedb') 
def index():
    print('欢迎来到京东主页')

@auth(auth_type='ldap')
def home(name):
    print('欢迎回家%s' %name)
#
@auth(auth_type='sssssss')
def shopping_car(name):
    print('%s的购物车里有［%s,%s,%s］' %(name,'奶茶','妹妹','娃娃'))


index()
home('产品经理')
shopping_car('产品经理')