在Python中,装饰器是一种函数或类,用于修改其他函数或类的功能。装饰器通过将函数或类作为参数传递给另一个函数,并返回一个新的函数或类来实现这一目的。这种技术在Python中被广泛应用于各种场景,如日志记录、性能监控、权限验证等。
首先,让我们来看一个简单的装饰器示例:
python
Copy Code
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 是一个装饰器函数,它接受一个函数作为参数,并返回一个新的函数 wrapper。当我们使用 @my_decorator 语法将装饰器应用到 say_hello 函数时,实际上相当于执行了 say_hello = my_decorator(say_hello)。调用 say_hello() 函数时,会先执行 wrapper 函数内的逻辑,然后再执行原始的 say_hello 函数。
除了简单的装饰器外,Python还支持带参数的装饰器。例如,我们可以编写一个带参数的装饰器来指定日志的级别:
python
Copy Code
def log(level):
def decorator(func):
def wrapper(args, **kwargs):
print(f"[{level}] {func.name} is called.")
return func(args, **kwargs)
return wrapper
return decorator
@log(level='INFO')
def say_hello(name):
print(f"Hello, {name}!")
say_hello("Alice")
在这个示例中,log 是一个带参数的装饰器工厂函数,它接受一个日志级别作为参数,并返回一个装饰器函数 decorator。decorator 函数接受一个函数作为参数,并返回一个新的函数 wrapper,用于添加日志记录功能。通过使用 @log(level='INFO') 将装饰器应用到 say_hello 函数上,我们可以指定日志级别为 INFO,从而在函数调用时记录相应的日志信息。
除了函数装饰器外,Python还支持类装饰器。类装饰器是指实现了 call 方法的类,它可以像函数装饰器一样被调用。下面是一个简单的类装饰器示例:
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.")
self.func(*args, **kwargs)
print("Something is happening after the function is called.")
@MyDecorator
def say_hello():
print("Hello!")
say_hello()
在这个示例中,MyDecorator 类实现了 init 和 call 方法,其中 init 方法用于接受被装饰的函数作为参数,并将其保存在实例变量 self.func 中,而 call 方法则用于实现装饰逻辑。通过使用 @MyDecorator 将类装饰器应用到 say_hello 函数上,我们可以在函数调用前后执行额外的逻辑。
总的来说,装饰器是Python中一种非常有用的工具,它可以帮助我们在不修改原有代码的情况下,为函数或类添加额外功能。通过合理地使用装饰器,我们可以提高代码的可读性和复用性,从而优化Python开发过程。