探秘Python里的装饰器

在Python的世界里，装饰器是一种强大而独特的工具，它能在不修改原有代码的基础上，为函数或类添加额外的功能。这一特性使得代码更加模块化、可维护性更高。

装饰器本质上是一个可调用对象，它接受一个函数作为参数，并返回一个新的函数。这个新函数通常会在调用原始函数之前或之后执行一些额外的操作。比如，我们可以使用装饰器来记录函数的执行时间。

import time

def timeit(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"{func.__name__} 执行时间：{end_time - start_time}秒")
        return result
    return wrapper

@timeit
def my_function():
    time.sleep(2)

my_function()

在上述代码中，timeit就是一个装饰器。当我们使用@timeit修饰my_function时，实际上是将my_function传递给timeit函数，然后将返回的wrapper函数赋值给my_function。

装饰器还可以用于验证函数的参数。例如，我们可以编写一个装饰器来确保函数的参数满足特定的条件。

def validate_args(func):
    def wrapper(*args):
        if all(isinstance(arg, int) for arg in args):
            return func(*args)
        else:
            print("参数必须是整数")
    return wrapper

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

add_numbers(1, 2)
add_numbers(1, "2")

装饰器还可以用于缓存函数的结果、实现权限验证等。通过使用装饰器，我们可以将这些通用的功能抽象出来，避免在每个函数中重复编写相同的代码。

Python的装饰器为我们提供了一种简洁而优雅的方式来扩展函数的功能。它不仅提高了代码的可读性和可维护性，还使得代码的复用性更强。深入理解和掌握装饰器的使用，将有助于我们编写更加高效、灵活的Python代码。