在Python中,装饰器是一种特殊类型的函数,它可以用来修改其他函数的行为。装饰器通过接收一个函数作为参数,然后返回一个新的函数,这个新函数通常会包含原始函数的功能,并添加一些额外的操作。
要实现功能扩展,你可以使用以下步骤:
@decorator_name
)来应用装饰器。下面是一个简单的示例,展示了如何使用装饰器来扩展一个函数的功能:
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
的装饰器,它接收一个函数func
作为参数。在装饰器内部,我们定义了一个名为wrapper
的内部函数,它首先打印一条消息,然后调用原始函数func
,最后再打印一条消息。最后,我们返回wrapper
函数。
要在say_hello
函数上应用这个装饰器,我们在函数定义之前使用@my_decorator
语法。当我们调用say_hello
时,实际上是在调用wrapper
函数,因此会看到所有三条消息都被打印出来。