Python默认装饰器

在Python中，装饰器（Decorator）是一种独特而强大的工具，它允许我们在不修改原有函数代码的情况下，为函数添加新的功能。装饰器的概念源于“开闭原则”（Open-Closed Principle），即软件实体（类、模块、函数等）应该对扩展开放，对修改关闭。本文将深入探讨Python中的默认装饰器及其使用方式。

什么是装饰器？

装饰器本质上是一个函数，它接受一个函数作为参数并返回一个新的函数。这个新的函数通常包含对原函数的调用，并可以在调用前后添加额外的功能。装饰器的使用通常通过@符号来实现，这使得代码更加简洁易读。

装饰器的使用形式

1. 直接使用

虽然这不是装饰器的标准用法，但了解其原理有助于深入理解装饰器。直接使用时，我们将被装饰的函数作为参数传递给装饰器函数，并接收返回的新函数。

def my_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

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

say_hello = my_decorator(say_hello)
say_hello()

2. 使用@符号

这是装饰器的标准用法。通过在函数定义前添加@符号和装饰器名称，我们可以更简洁地应用装饰器。

def my_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

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

say_hello()

3. 一个装饰器装饰多个函数

装饰器可以设计得足够灵活，以支持多个不同的函数。为了实现这一点，通常会在装饰器内部函数中使用*args和**kwargs来捕获任意数量的位置参数和关键字参数。

def my_decorator(func):
    def wrapper(*args, **kwargs):
        print("Something is happening before the function is called.")
        result = func(*args, **kwargs)
        print("Something is happening after the function is called.")
        return result
    return wrapper

@my_decorator
def add(x, y):
    return x + y

@my_decorator
def multiply(x, y):
    return x * y

print(add(3, 4))
print(multiply(3, 4))

自定义默认参数的装饰器

在某些情况下，我们可能想要创建一个装饰器，它能够自动将函数参数的默认值设置为参数的名称或其他值。虽然Python没有内置的“默认参数装饰器”，但我们可以通过使用inspect模块来手动实现这一功能。

from inspect import Parameter, Signature

def default_arguments(func):
    sig = Signature([
        Parameter(
            name=param.name,
            kind=param.kind,
            default=param.name if param.kind in (Parameter.POSITIONAL_OR_KEYWORD, Parameter.POSITIONAL_ONLY) else param.default
        )
        for param in Signature(func).parameters.values()
    ])

    def wrapper(*args, **kwargs):
        bound_arguments = sig.bind(*args, **kwargs)
        bound_arguments.apply_defaults()
        return func(*bound_arguments.args, **bound_arguments.kwargs)

    return wrapper

@default_arguments
def example(one, two, three):
    print(f"one={one}, two={two}, three={three}")

example()  # 输出: one=one, two=two, three=three
example("custom_one")  # 输出: one=custom_one, two=two, three=three

总结

Python的装饰器是一种非常有用的特性，它允许我们在不修改原函数代码的情况下，为函数添加额外的功能。通过了解装饰器的基本概念和使用方式，我们可以编写出更加灵活和强大的Python代码。无论是简单的日志记录、性能分析，还是复杂的权限控制、缓存管理，装饰器都可以成为我们手中的利器。