Python装饰器是一种设计模式,提供了一种在不改变原有函数代码的情况下,增强函数功能的方法。它们本质上是接受函数作为参数并返回新函数的高阶函数。
基本用法
我们从一个简单的例子开始,创建一个装饰器来记录函数执行的时间:
import time
def timing_decorator(func):
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
print(f"{func.__name__}运行时间: {end_time - start_time}秒")
return result
return wrapper
@timing_decorator
def example_function():
time.sleep(2)
example_function()
在这个例子中,timing_decorator
是一个装饰器,它接收一个函数func
,然后定义了一个包装函数wrapper
,在调用原始函数前后分别记录时间,最后输出执行时长。
创建自定义装饰器
要创建一个自定义装饰器,可以遵循以下步骤:
- 定义装饰器函数,它接受一个函数作为参数。
- 在装饰器函数内部,定义一个包装函数,这个包装函数会包含你想要添加的功能。
- 包装函数中调用原始的函数,并执行额外操作。
- 装饰器返回包装函数。
装饰带参数的函数
装饰器不仅可以用于无参数的函数,也可以用于带参数的函数。只需要确保包装函数接收任意数量的位置参数和关键字参数即可:
def my_decorator(func):
def wrapper(*args, **kwargs):
print("Before function call")
result = func(*args, **kwargs)
print("After function call")
return result
return wrapper
@my_decorator
def add(a, b):
return a + b
装饰类方法
装饰器还可以用于类方法。为了实现这一点,需要确保装饰器不会覆盖类的实例方法,这通常意味着在包装函数内部显式地传递self
参数:
class MyClass:
@staticmethod
def decorator(func):
def wrapper(self, *args, **kwargs):
print("Before method call")
result = func(self, *args, **kwargs)
print("After method call")
return result
return wrapper
@decorator
def my_method(self):
print("Inside the method")
总结
通过以上介绍,我们了解到装饰器不仅能够扩展函数的功能,而且还能保持代码的简洁性和模块化。无论是简单的日志记录、权限检查,还是复杂的资源管理,装饰器都提供了一种优雅的解决方案。掌握装饰器的使用,可以让你的代码更加Pythonic,同时也提高了开发效率。