叠加装饰器
叠加装饰器:
在同一个被装饰对象中,添加多个装饰器,并执行。
@装饰1
@装饰2
@装饰3
def 被装饰对象():
pass
注意:装饰器在调用被装饰对象时才会执行添加新功能。
- 叠加装饰器:
- 装饰的顺序: 由下到上装饰
- 执行的顺序: 由上往下
注意:无论inner中出现让任何的判断,最后的都要返回”调用后的装饰对象“func(*args, **kwargs)
def wrapper1(func):
def inner1(*args, **kwargs):
print('1---start')
res = func(*args, **kwargs)#被装饰对象在调用时,如果还有其他inner,会先执行其他装饰器中的inner
print('1---end')
return res
return inner1
def wrapper2(func):
def inner2(*args, **kwargs):
print('2---start')
res = func(*args, **kwargs)
print('2---end')
return res
return inner2
def wrapper3(func):
def inner3(*args, **kwargs):
print('3---start')
res = func(*args, **kwargs)
print('3---end')
return res
return inner3
@wrapper1 #inner1 = wrapper1(inner2)
@wrapper2 #inner2 = wrapper2(inner3)
@wrapper3 #inner3 = wrapper3(index)
def index():
print('from the index>>>>>')
index()
>>>>>>>>>
1---start
2---start
3---start
from the index>>>>>
3---end
2---end
1---end
装饰器的分类
# 无参装饰器: 装饰在被装饰对象时,没有传参数的装饰器。
'''
# 以下是无参装饰器
@wrapper1 # inner1 = wrapper1(inner2)
@wrapper2 # inner2 = wrapper2(inner3)
@wrapper3
'''
# 有参装饰器: 装饰被装饰的对象时,有传参的装饰器
在某些时候,我们需要给用户的权限进行分类
'''
# 以下是有参装饰器
@wrapper1(参数1) # inner1 = wrapper1(inner2)
@wrapper2(参数2) # inner2 = wrapper2(inner3)
@wrapper3(参数3)
'''
def user_auth(user_role):
'''
有参装饰器函数
'''
def wrapper(func):
def inner(*args, **kwargs):
if user_role == 'vip':
print('welcome!')
res = func(*args, **kwargs)
return res
elif user_role == '普通用户':
print('hi')
res = func(*args, **kwargs)
return res
return inner
return wrapper
@user_auth('普通用户') # wrapper = user_auth('普通用户')--->相当于@wrapper 语法糖
def ndex():
pass
index()
>>>>>>>
hi
wraps 的补充
wraps :是一个修复工具,修复的是被装饰对象的空间
from functools import wraps
def wrapper(func):
def inner(*args, **kwargs):
'''
此处是装饰器的注释
:param func:
:return:
'''
res = func(*args, **kwargs)
return res
return inner # ---》 func
@wrapper
def index():
'''
此处是index函数的注释
:return:
'''
pass
print(index) # 函数对象
# 函数对象.__doc__: 查看函数内部的注释
print(index.__doc__) # inner.__doc__
>>>>>>>
<function wrapper.<locals>.inner at 0x000002A0A52A7C80>
此处是装饰器的注释
:param func:
:return:
下面是使用了wraps后的 修复被装饰对象空间,下面是被装饰对象的注释
def wrapper(func):
@wraps(func) # 修改名称空间: inner ---》 func 此处修改成功
def inner(*args, **kwargs):
'''
此处是装饰器的注释
:param func:
:return:
'''
res = func(*args, **kwargs)
return res
return inner # ---》 func
@wrapper
def index():
'''
此处是index函数的注释
:return:
'''
pass
print(index) # 函数对象
# 函数对象.__doc__: 查看函数内部的注释
print(index.__doc__) # inner.__doc__
>>>>>>>>>>>>
<function index at 0x00000199BCC77C80>
此处是index函数的注释
:return:
编写装饰器 为多个函数加上认证的功能(用户的账号i密码来源于文件)要求登录一次,后续的函数无需再输入用户名和密码。
# 可变类型,无需通过global引用,可直接修改
user_info = {
'user': None # username
}
# 不可变类型,需要在函数内部使用global对其进行修改
user = None
def login():
# 判断用户没有登录时,执行
# 登录功能
# global user
username = input('请输入账号: ').strip()
password = input('请输入密码: ').strip()
with open('user.txt', 'r', encoding='utf-8') as f:
for line in f:
print(line)
name, pwd = line.strip('
').split(':') # [tank, 123]
if username == name and password == pwd:
print('登录成功!')
user_info['user'] = username
else:
print('登录失败!')
def login_auth(func): # func ---> func1, func2, func3
def inner(*args, **kwargs):
# 已经登录,将被装饰对象直接调用并返回
if user_info.get('user'):
res = func(*args, **kwargs) # func() ---> func1(), func2(), func3v
return res
# 若没登录,执行登录功能
else:
print('请先登录...')
login()
return inner
# func1,2,3 都需要先登录才可以使用,若登录一次,
# 后续功能无需再次登录,绕过登录认证
@login_auth
def func1():
print('from func1')
pass
@login_auth
def func2():
print('from func2')
pass
@login_auth
def func3():
print('from func3')
pass
while True:
func1()
input('延迟操作...')
func2()
func3()
>>>>>>>>>>>>>
请输入你的账户:david
请输入你的密码:123
登录成功
{'user': 'david'}
from func1
延迟操作
from func2
from func3
from func1
延迟操作
# 无参装饰器模板
def wrapper(func):
def inner(*args, **kwargs):
res = func(*args, **kwargs)
# 在被装饰对象调用后添加功能
return res
return inner
# 有参装饰器模板
def outer(arg):
def wrapper(func):
def inner(*args, **kwargs):
res = func(*args, **kwargs)
return res
return inner
return wrapper