Python装饰器参数的获取方法

在Python编程中，装饰器是一项强大的功能，它允许我们在不修改原有函数代码的情况下，为函数添加额外的功能。而获取装饰器的参数，能进一步增强装饰器的灵活性和实用性。

简单回顾一下装饰器的基本概念。装饰器本质上是一个函数，它接收一个函数作为参数，并返回一个新的函数。例如：

def my_decorator(func):
    def wrapper():
        print("在函数执行前执行的代码")
        func()
        print("在函数执行后执行的代码")
    return wrapper

@my_decorator
def say_hello():
    print("Hello!")

这里my_decorator就是一个装饰器，say_hello函数被装饰后，执行say_hello时会先打印“在函数执行前执行的代码”，然后打印“Hello!”，最后打印“在函数执行后执行的代码”。

当我们需要给装饰器传递参数时，情况会稍微复杂一些。我们可以定义一个带参数的装饰器工厂函数。例如：

def repeat(n):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(n):
                result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator

@repeat(3)
def greet(name):
    print(f"Hello, {name}!")

在这个例子中，repeat是一个装饰器工厂函数，它接收参数n。repeat返回一个装饰器decorator，这个装饰器再接收被装饰的函数func。wrapper函数会根据n的值重复执行func。

还有一种情况是装饰器带多个参数。我们可以这样实现：

def log(message, level='INFO'):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print(f"[{level}] {message} before function {func.__name__}")
            result = func(*args, **kwargs)
            print(f"[{level}] {message} after function {func.__name__}")
            return result
        return wrapper
    return decorator

@log("Function call", level='DEBUG')
def add(a, b):
    return a + b

在上述代码中，log装饰器工厂函数接收message和level两个参数，在wrapper函数中利用这些参数进行日志打印。

通过合理地获取和利用装饰器的参数，我们能够根据不同的需求定制装饰器的行为，让代码更加简洁、高效且易于维护。掌握装饰器参数的获取方法，无疑为Python开发者在代码优化和功能扩展方面提供了有力的工具。