Python中的装饰器如何传递参数？

深入理解Python装饰器中的参数传递机制

引言与核心答复

在Python编程中,装饰器（Decorators）作为强大的元编程工具，允许你在不修改原函数代码的基础上，动态地添加功能，当装饰器本身需要接受参数时，许多开发者可能会感到困惑，本文将详细解析如何在Python装饰器中有效传递参数，通过示例演示其工作机制，并探讨一些最佳实践，帮助您更好地掌握这一高级特性。

Python中的装饰器如何传递参数？

装饰器基础回顾

回顾一下装饰器的基本概念,一个简单的装饰器通常定义为一个函数，它接受一个函数（或方法）作为参数，并返回一个新的函数，这个新函数通常会在执行原函数前后添加额外的逻辑。

def simple_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
@simple_decorator
def say_hello():
    print("Hello!")

在这个例子中,@simple_decorator语法糖使得say_hello函数被simple_decorator装饰，执行时会先打印前导信息，再执行原函数，最后打印后续信息。

装饰器传递参数的需求

当装饰器需要根据不同配置执行不同行为时,就需要向装饰器传递参数，你可能希望根据参数决定是否记录日志、设置不同的超时时间等，这时，我们需要构建一个能够接受参数的装饰器工厂函数。

实现接受参数的装饰器

为了使装饰器能接受参数,你需要创建一个外层函数，该函数接受装饰器的参数，并返回一个实际的装饰器函数，这个返回的装饰器函数再按照常规方式接受一个函数作为参数，并返回包装后的函数，下面是一个示例：

def repeat_rule(num_times):
    def decorator_repeat(func):
        def wrapper(*args, **kwargs):
            for _ in range(num_times):
                value = func(*args, **kwargs)
            # 可以选择返回最后一次调用的结果，或者根据需求调整
            return value  # 或省略，如果函数不返回值
        return wrapper
    return decorator_repeat
@repeat_rule(num_times=3)
def greet(name):
    print(f"Hello {name}!")

在这个例子中,repeat_rule是一个装饰器工厂函数，它接受num_times作为参数，返回decorator_repeat装饰器。decorator_repeat再按照常规装饰器的方式工作，包装greet函数，使其被调用三次。