在Python中使用装饰器对函数进行扩展和修饰

装饰器是Python中一种强大且常用的机制，可以在不改变原有函数定义的情况下，对函数进行扩展和修饰。装饰器通常用于为函数添加额外的功能，比如日志、参数验证、缓存等。

为了理解装饰器的使用方式和原理，我们先从一个简单的例子开始。

def log(func):
    def wrapper(*args, **kwargs):
        print('Calling function:', func.__name__)
        return func(*args, **kwargs)
    return wrapper

@log
def add(a, b):
    return a + b

result = add(2, 3)
print('Result:', result)

在上面的例子中，我们定义了一个名为log的装饰器函数。装饰器函数的参数是被修饰的函数，它内部定义了一个名为wrapper的函数，用于对被修饰的函数进行扩展和修饰。在wrapper函数中，我们先打印出被修饰函数的名称，然后再调用被修饰函数。最后，装饰器函数返回wrapper函数。

接下来，我们通过@符号将装饰器应用到add函数上。相当于执行了以下代码：

add = log(add)

当我们调用add函数时，实际上调用的是被修饰后的函数wrapper。在wrapper函数中，我们先打印出'Calling function: add'，然后再调用原来的add函数。最后，返回add函数的执行结果。

最终的输出结果为：

Calling function: add
Result: 5

可以看到，装饰器起到了在调用add函数之前和之后打印日志的作用。这个例子展示了装饰器的基本用法和原理。

除了上述的日志功能，装饰器还可以用于其他各种功能扩展，下面列举几个例子：

1. 参数验证：可以使用装饰器对传入函数的参数进行验证，确保参数的有效性和完整性。

def validate_args(func):
    def wrapper(*args, **kwargs):
        if len(args) != 2:
            raise ValueError('Two arguments are required')
        if not all(isinstance(arg, int) for arg in args):
            raise TypeError('Arguments must be integers')
        return func(*args, **kwargs)
    return wrapper

@validate_args
def add(a, b):
    return a + b

2. 缓存：可以使用装饰器来实现函数调用的结果缓存，提高函数的执行效率。

def cache(func):
    cache_dict = {}

    def wrapper(*args, **kwargs):
        key = tuple(args) + tuple(kwargs.items())
        if key in cache_dict:
            return cache_dict[key]
        result = func(*args, **kwargs)
        cache_dict[key] = result
        return result

    return wrapper

@cache
def calculate(a, b):
    print('Performing expensive calculation...')
    return a + b

3. 计时：可以使用装饰器来统计函数的执行时间，以便进行性能分析和优化。

import time

def timing(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print('Execution time:', end_time - start_time)
        return result
    return wrapper

@timing
def add(a, b):
    return a + b

通过使用装饰器，我们可以方便地对函数进行扩展和修饰，而不需要修改函数本身的定义。这种方式使得代码更加模块化、可复用和易于维护。同时，装饰器的使用也提高了代码的可读性和可维护性。

总结来说，装饰器是Python中一种强大而常用的机制，用于对函数进行扩展和修饰。通过定义装饰器函数和使用@符号将装饰器应用到函数上，我们可以方便地为函数添加额外的功能，比如日志、参数验证、缓存等。装饰器的使用使得代码更加模块化、可复用和易于维护，提高了代码的可读性和可维护性。