利用inspect模块获取Python装饰器传入参数的方法

在Python编程中，装饰器是一项强大的功能，它允许我们在不修改原有函数代码的基础上，为函数添加额外的功能。然而，有时我们需要获取装饰器传入的参数，以便进行更灵活的操作。这时，inspect模块就能发挥重要作用。

我们要了解inspect模块。它提供了一系列用于获取对象信息的函数，比如获取函数参数、局部变量等。对于处理装饰器参数来说，它是一个得力工具。

来看一个简单的示例。假设我们有一个装饰器，用于在函数执行前后打印一些日志信息，并且希望这个装饰器能接收参数来定制日志内容。代码如下：

import inspect


def log_decorator(prefix):
    def wrapper(func):
        def inner(*args, **kwargs):
            sig = inspect.signature(func)
            bound_args = sig.bind(*args, **kwargs)
            bound_args.apply_defaults()
            print(f"{prefix} 函数 {func.__name__} 即将执行，参数为: {bound_args.arguments}")
            result = func(*args, **kwargs)
            print(f"{prefix} 函数 {func.__name__} 执行完毕")
            return result
        return inner
    return wrapper


@log_decorator(prefix="INFO")
def add_numbers(a, b):
    return a + b

在上述代码中，log_decorator 是一个装饰器函数，它接收一个参数 prefix。在内部的 inner 函数中，我们使用了 inspect.signature 来获取被装饰函数 func 的签名信息。通过 sig.bind 方法将实际传入的参数绑定到函数签名上，并使用 apply_defaults 方法应用默认参数。这样，我们就能通过 bound_args.arguments 获取到函数调用时的所有参数，包括位置参数和关键字参数。

通过这种方式，我们不仅能清晰地知道装饰器传入的参数，还能在函数执行前后根据这些参数进行定制化的日志输出或其他操作。利用inspect模块获取Python装饰器传入参数，能够让我们的代码更加灵活和可维护，无论是在开发小型脚本还是大型项目中，都能为我们带来很大的便利。掌握这一技巧，将有助于我们在Python编程中更高效地实现各种功能需求。