Python装饰器如何简化代码 - 问答

Python装饰器是一种在不修改原始函数代码的情况下，为函数增加新功能的方法。它们通过在原始函数前添加一个包装函数来实现这一目的。装饰器可以简化代码，提高代码的可读性和可维护性。

以下是一个简单的装饰器示例：

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()

在这个例子中，my_decorator 是一个装饰器函数，它接受一个函数 func 作为参数。wrapper 函数是实际的包装函数，它在调用原始函数之前和之后执行一些额外的操作。通过使用 @my_decorator 语法，我们将 say_hello 函数与 my_decorator 装饰器关联起来。

当调用 say_hello() 时，实际上是在调用 wrapper 函数，因此会看到以下输出：

Something is happening before the function is called.
Hello!
Something is happening after the function is called.

装饰器可以简化代码，例如：

def log_time(func):
    import time
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"{func.__name__} took {end_time - start_time:.2f} seconds to execute.")
        return result
    return wrapper

@log_time
def my_function():
    time.sleep(2)
    print("Function executed.")

my_function()

在这个例子中，log_time 装饰器用于记录函数的执行时间。通过使用装饰器，我们可以轻松地为其他函数添加相同的日志功能，而无需修改它们的代码。

0 赞

0 踩