如何在 Python 中创建和链接函数装饰器？

如何创建函数装饰器并将它们链接在一起

简介

Python 中的装饰器允许您可以修改函数的行为而不修改函数本身。您可以使用装饰器在调用函数之前和之后添加功能、检查输入/输出以及执行其他操作。

创建装饰器

创建装饰器，定义一个将另一个函数作为参数的函数。此函数应返回一个包装原始函数的新函数。

def make_bold(fn):
    # The new function the decorator returns
    def wrapper():
        # Insertion of some code before and after
        return "<b>" + fn() + "</b>"
    return wrapper

链接装饰器

您可以链接装饰器以对函数应用多个修改。只需依次应用每个装饰器，如下所示：

@make_bold
@make_italic
def say():
    return "hello"

此代码将 make_bold 装饰器应用到 make_italic 装饰器，后者又应用到 say 函数。结果是 say 函数返回一个粗体和斜体的字符串。

示例：链接装饰器以格式化文本

假设您有一个返回 a 的函数文本字符串。您想要向文本添加粗体和斜体格式。您可以使用上面定义的装饰器来实现此目的：

@make_bold
@make_italic
def formatted_text(text):
    return text

print(formatted_text("Hello world"))

输出：

<b><i>Hello world</i></b>

将参数传递给装饰器

您可以通过将参数括在中来将参数传递给装饰器括号：

def a_decorator_passing_arguments(function_to_decorate):
    def a_wrapper_accepting_arguments(arg1, arg2):
        print("I got args! Look: {0}, {1}".format(arg1, arg2))
        function_to_decorate(arg1, arg2)
    return a_wrapper_accepting_arguments

@a_decorator_passing_arguments
def print_full_name(first_name, last_name):
    print("My name is {0} {1}".format(first_name, last_name))

print_full_name("Peter", "Venkman")

输出：

I got args! Look: Peter, Venkman
My name is Peter Venkman

结论

通过了解如何创建和链接装饰器，您可以扩展 Python 函数的功能。这种强大的技术允许您以模块化和可重用的方式修改行为、执行检查并增强代码。

以上是如何在 Python 中创建和链接函数装饰器？的详细内容。更多信息请关注PHP中文网其他相关文章！