python 多层装饰器分析

发布时间：2018-04-04 00:11:50编辑：admin阅读（4756）

装饰器本质上是一个Python函数，它可以让其他函数在不需要做任何代码变动的前提下增加额外功能，装饰器的返回值也是1函数对象。它经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计，有了装饰器，我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。概括的讲，装饰器的作用就是为已经存在的对象添加额外的功能。




一层装饰器

应用场景最多的就是登录判断

举个例子

默认情况下,登录状态是False。现在有一个需求，需要执行func1执行时，必须要登录才行。

需要增加一个装饰器用来判断

#用户状态(全局变量)
login_state = {'username':None,'status':False}

#装饰器模板
def first(f):
    def inner(*args,**kwargs):
        '''被装饰之前'''
        ret = f(*args, **kwargs)
        '''被装饰之后'''
        return ret
    return inner

#测试函数
def func1(*args,**kwargs):
    print(args,kwargs)
    return 666
#执行函数
print(func1())

执行输出：

() {}

666




代码如下：


#用户状态(全局变量)
login_state = {'username':None,'status':False}

#假设用户名存在文件或者数据库中
info = {'username':'xiao','password':'123'}
#失败次数
failures = 1
#最大登录次数
maximum = 3

#装饰器模板
def first(f):
    def inner(*args,**kwargs):
        '''被装饰之前'''
        if login_state['status']:
            ret = f(*args, **kwargs)
        else:
            print('您还未登录,请先登录账户!')
            #执行登录程序
            result = login()
            #判断结果
            if result:
                print('登录成功')
            else:
                print('登录失败')
            ret = f(*args, **kwargs)

        #执行被装饰的函数
        return ret
        '''被装饰之后'''

    return inner

#登录判断逻辑
def login():
    global failures
    while failures <= maximum:
        username = input('请输入用户: ').strip()
        password = input('请输入密码: ').strip()
        #判断用户名和密码是否一致
        if username == info['username'] and password == info['password']:
            # 修改全局变量,这里不用global,因为它是可变类型
            login_state['username'] = username
            login_state['status'] = True
            result = True
            break
        else:
            print('您输入的用户或者密码错误，请重新输入，还有%s次机会' % (maximum - failures))
            failures += 1
            result = False
    return result

#测试函数
@first
def func1(*args,**kwargs):
    print(args,kwargs)
    return 666

#执行函数
print(func1())

执行输出：






现在又有一个需求，执行watch_movie函数时，必须是VIP用户才可以。


增加二层装饰器

#!/usr/bin/env python
# coding: utf-8
__author__ = 'www.py3study.com'

#用户状态(全局变量)
login_state = {'username':None,'status':False}

#假设用户名存在文件或者数据库中
info = {'username':'xiao','password':'123','level':None}
#失败次数
failures = 1
#最大登录次数
maximum = 3

#第一层装饰器,判断登录
def first(f):
    def inner(*args,**kwargs):
        '''被装饰之前'''
        if login_state['status']:
            ret = f(*args, **kwargs)
        else:
            print("\033[41;1m您还未登录,请先登录账户!\033[0m")
            #执行登录程序
            result = login()
            #判断结果
            if result:
                print('登录成功')
            else:
                print("\033[41;1m登录失败!\033[0m")

            ret = f(*args, **kwargs)

        #执行被装饰的函数
        return ret
        '''被装饰之后'''

    return inner

#第二层装饰器,判断VIP
def second(f):
    def inner(*args,**kwargs):
        '''被装饰之前'''
        #判断电影名后缀是否包含VIP
        if args[0].endswith('VIP'):
            if info['level'] == None:
                return "\033[41;1m该影片只对VIP用户开放!\033[0m"
            else:
                ret = f(*args, **kwargs)
        else:
            #免费电影
            ret = f(*args, **kwargs)

        return ret
        '''被装饰之后'''
        #return ret
    return inner

#登录判断逻辑
def login():
    global failures
    while failures <= maximum:
        username = input('请输入用户: ').strip()
        password = input('请输入密码: ').strip()
        #判断用户名和密码是否一致
        if username == info['username'] and password == info['password']:
            # 修改全局变量,这里不用global,因为它是可变类型
            login_state['username'] = username
            login_state['status'] = True
            #设置标记
            result = True
            break
        else:
            print("\033[41;1m您输入的用户或者密码错误，请重新输入，还有%s次机会!\033[0m" % (maximum - failures))
            failures += 1
            result = False

    return result

#看电影函数
@first
@second
def watch_movie(name):
    if name.strip() == '':
        return '电影名不能为空'
    return '正在播放影片 {}'.format(name)

#首页
def index():
    movie_list = {
        #序号:电影名
        1: '黑暗塔',
        2: '环太平洋 VIP',
        3: '变形金刚5:最后的骑士 VIP',
        4: '退出',
    }

    while True:
        #循环列表
        for k, v in movie_list.items():
            print('{}\t{}'.format(k, v))
        choice_num = input('请选择输入的序号:').strip()
        if choice_num.isdigit():
            choice_num = int(choice_num)
            #判断数字范围
            if 0 < choice_num < len(movie_list):
                #执行看电影函数
                print(watch_movie(movie_list[choice_num]))
            elif choice_num == len(movie_list):
                break
            else:
                print("\033[41;1m请输入范围内的序号!\033[0m")
        else:
            print("\033[41;1m您输入的有非法字符，请重新输入!\033[0m")

if __name__ == "__main__":
    index()

执行输出：



修改全局变量

info = {'username':'xiao','password':'123','level':'VIP'}

再次登录，就可以看VIP电影了




装饰器执行顺序是这样的



上述代码的二层装饰器，都是在被装饰之前操作的

first判断是否登录

second判断用户等级




二层以上装饰器，也都是按照上图的循环来的，多套了几层而已！

关键字：