正文
python3装饰器用法示例
提示：扫一扫查出行【扫一扫了解最新限行尾号】
复制提示
装饰器在编写后台的逻辑时有可能会用到，比方说一个场景：公司的员工想要登录自己公司的考勤记录系统去修改自己的考勤，以前是随便谁都有权限去修改，这样老板不同意了，现在，要在你登录前加一个权限验证的逻辑，如果你权限没有达到等级要求，你就不能登录系统去修改自己的考勤记录。如何在不改动原函数的情况下去增加这部分权限验证的功能呢？这就用到的装饰器函数。
装饰器函数的一大有点就是，在不改动原函数的基础上，去给函数增加新的功能。
装饰器有两种，一种是没有参数的，一种是有参数的，本示例两种装饰器都有，不过只运行了有参数的。
有关用法和解释在代码中已经注释。下面是一个例子。
# 装饰器函数的两个特点，也是必须满足的条件
# 1. 闭包
# 2. 外部函数的返回值必须是内部函数# 这是一个没有参数的装饰器的例子
def decorator(func):
def wrapper():
print('在登录前，先检查有没有此账号。')
# !@#$%^&*，在数据库中查询
# 如果数据库的已注册表中查到此用户，就执行登录函数 login()。
# 如果没有查到，则显示“请先注册，然后登录”
func()
return wrapper# 上面已经写好了一个装饰器，下面是用法
@decorator
def login1():
# 比方说这里是一段很复杂的逻辑，现在用登录操作代替
print('我现在要登录了')# 这是一个有参数的装饰器例子
def decorator2(func):
def wrapper2(name):
# 数字越小，权限越高
priLevelDict = {'赵总': 1, '赵总': 2, '小王': 3} # 判断字典中是否存在某个key，可以用 in 或者 not in来判断
# print('小王' in priLevelDict) if name in priLevelDict:
print('正在对{}进行权限验证'.format(name))
# 如果权限验证通过，则可以进行下一步的登录，管理
# 如果权限验证不通过，则不能登录。逻辑自己设计就好
# print(priLevelDict[name])
if priLevelDict[name] == 1 or priLevelDict[name] == 2:
print('验证通过')
func(name)
elif priLevelDict[name] == 3:
print('对不起，{}，您没有权限登录，请有点自知之明。'.format(name))
print()
else:
print('你tm是谁啊，数据库里都找不到你，滚犊子。') return wrapper2@decorator2
def login2(name):
print('尊敬的{}， 现在您可以登录管理系统进行管理'.format(name))
print()if __name__ == '__main__':
while 1:
name = input('请输入你的姓名：')
login2(name)
运行结果：
请输入你的姓名：赵总
正在对赵总进行权限验证
验证通过
尊敬的赵总， 现在您可以登录管理系统进行管理
# 这可是领导，老板要查岗，必须得通过啊请输入你的姓名：钱经理
正在对钱经理进行权限验证
验证通过
尊敬的钱经理， 现在您可以登录管理系统进行管理
# 这是顶头上司，也要好好巴结一下请输入你的姓名：小王
正在对小王进行权限验证
对不起，小王，您没有权限登录，请有点自知之明。
# 这是同事，说话比较随意请输入你的姓名：老李
你tm是谁啊，数据库里都找不到你，滚犊子。
# 这不是我们公司员工 ，不用客气
请输入你的姓名：
