装饰器的介绍

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

原则：
1 不修改被修饰函数的源代码
2 不修改被装饰函数的调用方式

装饰器的知识储备
装饰器=高阶函数+函数嵌套+闭包

高阶函数的定义：
1 函数接收的参数是一个函数名
2 函数的返回值是一个函数名
3 满足上述条件之一 就可以称为一个高阶函数

函数嵌套：
在函数里面定义函数

高阶函数

import time
def foo():
    time.sleep(1)
    print('hello,world')

def timer(func):
    start_time = time.time()
    func()
    stop_time = time.time()
    print('函数运行的时间为：%s'%(stop_time-start_time))
    return func

foo = timer(foo)
foo()

 

装饰器的框架：

import time
def timer (func):
    def wrapper():
        print(func)
        start_time = time.time()
        res = func()
        stop_time = time.time()
        print('test函数运行的时间是：%s'%(stop_time-start_time))
        return res
    return wrapper
@timer
def test():
    time.sleep(3)
    print('test函数运行完毕')
    return 'test的返回值'

# test = timer(test) #wrapper的地址    @timer 相当于  test = timer(test)
ret = test() #运行wrapper函数
print(ret)

 

函数闭包加上认证功能：

user_list = [
    {'name': 'ball', 'password': '123'},
    {'name': 'baby', 'password': '123'},
    {'name': 'alex', 'password': '123'},
]
current_dic = {'username': None, 'login': False}  # 当前的登录状态

def auth_func(func):
    def wrapper(*args, **kwargs):
        if current_dic['username'] and current_dic['login']: #有用户名登录 执行函数
            res = func(*args, **kwargs)
            return res
        username = input('请输入用户名：').strip()  #没有用户名  需登录
        password = input('请输入密码：').strip()
        for user_dic in user_list:   #判断登录的用户名是列表内的用户
            if username == user_dic['name'] and password == user_dic['password']:
                current_dic['username'] = username    #修改当前登录状态
                current_dic['login'] = True
                res = func(*args, **kwargs)
                return res
        else:
            print('用户名或者密码错误，请重新输入！')
    return wrapper


@auth_func
def index():
    print('欢迎来到淘宝首页')


@auth_func
def shopping_car(name):
    print('%s的购物车里有%s%s' % (name, '苹果', '芒果'))


@auth_func
def order():
    pass


print('befor--->', current_dic)
index()
print('after-->', current_dic)
shopping_car('nihao')

 

带参函数认证：

user_list = [
    {'name': 'ball', 'password': '123'},
    {'name': 'baby', 'password': '123'},
    {'name': 'alex', 'password': '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']: #有用户名登录 执行函数
                    res = func(*args, **kwargs)
                    return res
                username = input('请输入用户名：').strip()  #没有用户名  需登录
                password = input('请输入密码：').strip()
                for user_dic in user_list:   #判断登录的用户名是列表内的用户
                    if username == user_dic['name'] and password == user_dic['password']:
                        current_dic['username'] = username    #修改当前登录状态
                        current_dic['login'] = True
                        res = func(*args, **kwargs)
                        return res
                else:
                    print('用户名或者密码错误，请重新输入！')
            elif auth_type == 'qita':
                print('不知道什么类型')
        return wrapper
    return auth_func


@auth(auth_type='filedb')
def index():
    print('欢迎来到淘宝首页')


@auth(auth_type='filedb')
def shopping_car(name):
    print('%s的购物车里有%s%s' % (name, '苹果', '芒果'))


@auth(auth_type='filedb')
def order():
    pass


print('befor--->', current_dic)
index()
print('after-->', current_dic)
shopping_car('nihao')