文章目录
概要
装饰器来自decorator的直译。什么叫装饰,就是装点、提供一些额外的功能;
装饰器本质上是一个Python函数(其实就是闭包),它可以让其他函数在不需要做任何代码变动的前提下增加额外功能,装饰器的返回值也是一个函数对象.
例如:
经典的应用有插入日志、增加计时逻辑来检测性能、加入事务处理等
一、函数定义装饰器(装饰器不带参数)
def decorator(obj):
def wrapper(*args,**kwargs):
'''
函数里面可以操作自己想加入的功能;
加上*args,**kwargs两个参数,即表示:
无论被修饰的函数/类有几个参数、什么类型,都可以用该装饰器装饰,
如果,没有写这两个参数,那就必须被装饰的函数/类有
几个参数,装饰器就写死为几个参数;就达不到想要的效果
'''
print("执行装饰器")
res=obj(*args,**kwargs)
return res
return wrapper
#装饰器可装饰函数:
@decorator
def add(x,y):
return x+y
二、函数定义装饰器(装饰器带参数)
# 举例两个变量
user = 'admin'
pwd = 'admin123'
# 装饰器包含3层,最外层接收装饰器参数auth_type,第二层outter,接收装饰的函数func,其余与装饰器无异
def auth(auth_type):
def outter(func):
def wrapper(*args, **kwargs):
if auth_type == 'local':
username = input("Enter your username: ").strip()
password = input("Enter your password: ").strip()
if user == username and pwd == password:
print("Access granted")
else:
print("Access denied")
else:
print('不用验证')
return func()
return wrapper
return outter
# 一般的装饰器不带括号与参数,带参数时一般逻辑更为复杂,如:装饰器中需要根据参数不同,走不同逻辑
@auth(auth_type='local1')
def index():
print('welcome to the index page')
return 'index'
三、装饰器装饰同一个类里的函数
提示:from functools import wraps
from functools import wraps
class Test(object):
def __init__(self):
self.reset = True
self.flag = True
# 在类里定义一个装饰器
def info(func):
@wraps(func)
def wrapper(self, *args, **kwargs):
print('log_type: info'.center(50, '-'))
if self.reset:
print('Reset is Ture, change flag...')
self.flag = False
return func(self, *args, **kwargs)
return wrapper
@info
def speak(self, text):
print(f'speak: {text}')
if __name__ == '__main__':
t = Test()
t.speak('hello world')
print(t.flag)
3、1 效果展示:举例:打印日志–》被装饰函数的注释:
四、类装饰器
提示:这里可以添加总结
class Test:
def __init__(self, func):
self.func = func
def __call__(self, *args, **kwargs):
print('函数__call__被调用了'.center(50, '-'))
return self.func(*args, **kwargs)
@Test
def speak(text):
print(f'speak: {text}')
if __name__ == '__main__':
speak('hello world')