Python装饰器是一种非常有用的功能,它允许用户在不修改原有函数代码的情况下增加额外的功能。装饰器可以应用于函数或方法上,用于日志记录、访问控制、度量、性能测试、事务处理、权限校验、缓存等各种场景,还有更多关于Python装饰器的高级用法和特殊场景,包括带返回值的装饰器、装饰器应用在协程上,以及使用装饰器来管理状态或者实现设计模式等。
文章目录
1. 基本装饰器
基本装饰器是最简单的装饰器形式,它用于在函数执行前后添加功能。
示例:一个简单的日志记录装饰器。
def simple_decorator(func):
def wrapper():
print(f"Function {
func.__name__} is about to be executed.")
func()
print(f"Function {
func.__name__} has finished execution.")
return wrapper
@simple_decorator
def say_hello():
print("Hello!")
say_hello()
2. 带参数的装饰器
当需要向装饰器传递额外参数时,可以定义一个返回装饰器的装饰器。
示例:一个接受参数的日志记录装饰器。
def logging_decorator(level="INFO"):
def decorator(func):
def wrapper(*args, **kwargs):
if level == "INFO":
print(f"[INFO] Executing {
func.__name__}")
elif level == "DEBUG":
print(f"[DEBUG] Executing {
func.__name__} with args: {
args}, kwargs: {
kwargs}")
result = func(*args, **kwargs)
print(f"[INFO] Finished {
func.__name__}")
return result
return wrapper
return decorator
@logging_decorator(level="DEBUG")
def add(a, b):
return a + b
print(add(5, 3))
3. 装饰类的方法
装饰器也可以用于类的方法上,例如用于校验权限、修改方法行为等。
示例:检查用户是否有执行方法的权限。
def check_permission(permission):
def decorator(func):
def wrapper(self, *args