温馨提示×

Python中怎么使用装饰器来装饰函数

小亿
82
2024-03-12 15:22:44
栏目: 编程语言

在Python中,装饰器是一种函数,它接受一个函数作为参数,并返回一个新的函数。通过使用装饰器,可以在不修改原始函数代码的情况下添加额外的功能或逻辑。

下面是一个简单的示例,演示如何使用装饰器来装饰函数:

# 定义一个装饰器函数
def my_decorator(func):
    def wrapper():
        print("Before calling the function")
        func()
        print("After calling the function")
    return wrapper

# 使用装饰器来装饰函数
@my_decorator
def say_hello():
    print("Hello!")

# 调用装饰后的函数
say_hello()

在上面的示例中,my_decorator是一个装饰器函数,它接受一个函数作为参数,并返回一个新的函数wrapper。在wrapper函数中,我们可以在调用原始函数之前和之后添加额外的逻辑。通过在say_hello函数上方添加@my_decorator语法,我们将say_hello函数传递给my_decorator装饰器函数,并将返回的新函数赋值给say_hello,这样say_hello函数就被装饰了。

当我们调用say_hello函数时,实际上会调用装饰后的wrapper函数,从而在打印"Hello!“之前和之后分别打印"Before calling the function"和"After calling the function”。这样,我们就成功地使用装饰器来装饰函数。

0