在 Python 中,装饰器(decorator)是一种高级功能,用于修改或增强函数或方法的行为。装饰器本质上是一个函数,它接收另一个函数作为参数,并返回一个新的函数或修改后的函数。装饰器常用于代码复用、日志记录、性能测试等场景。
基本概念
1. 装饰器的基本语法
装饰器的基本结构如下:
def my_decorator(func):
def wrapper():
# 在函数调用之前执行的代码
print("Something is happening before the function is called.")
func()
# 在函数调用之后执行的代码
print("Something is happening after the function is called.")
return wrapper
@my_decorator
def say_hello():
print("Hello!")
say_hello()
在这个例子中,@my_decorator 是一个装饰器的应用,它会在调用 say_hello 函数之前和之后分别打印一些内容。
2. 装饰器的工作原理
装饰器的工作原理可以分为以下几个步骤:
定义装饰器函数:装饰器是一个函数,它接收一个函数作为参数,并返回一个新的函数。
应用装饰器:使用 @decorator_name 的语法,将装饰器应用到目标函数上。这个过程实际上是将目标函数传递给装饰器函数,并用装饰器函数返回的新函数替代原来的函数。
调用装饰后的函数:装饰器返回的函数可以在原函数之前或之后执行额外的代码。
常见的装饰器模式
1. 带参数的装饰器
如果需要让装饰器接受参数,可以将装饰器再封装一层:
python
Copy Code
def repeat(num_times):
def decorator_repeat(func):
def wrapper(*args, **kwargs):
for _ in range(num_times):
func(*args, **kwargs)
return wrapper
return decorator_repeat
@repeat(num_times=3)
def greet(name):
print(f"Hello {name}")
greet("Alice")
在这个例子中,repeat 装饰器接受一个参数 num_times,表示要重复调用被装饰的函数的次数。
2. 装饰器的类实现
装饰器不仅可以用函数实现,还可以用类实现:
python
Copy Code
class MyDecorator:
def __init__(self, func):
self.func = func
def __call__(self, *args, **kwargs):
print("Something is happening before the function is called.")
result = self.func(*args, **kwargs)
print("Something is happening after the function is called.")
return result
@MyDecorator
def say_hello():
print("Hello!")
say_hello()
在这个例子中,MyDecorator 是一个装饰器类,它的 __call__ 方法定义了装饰器的行为。
内置装饰器
Python 标准库中也提供了一些内置装饰器,例如:
@staticmethod:将方法定义为静态方法。
@classmethod:将方法定义为类方法。
@property:将方法定义为属性。
装饰器的实际应用
装饰器在实际开发中非常有用,例如:
日志记录:在函数调用时自动记录日志。
权限验证:检查用户是否有权限执行某些操作。
性能测试:测量函数执行的时间。
缓存:缓存函数的结果以提高性能。
总之,装饰器是 Python 中一个强大的工具,能够帮助开发者以更优雅和模块化的方式扩展和修改函数的行为。