使用装饰器实现Python函数的重试机制

在编写函数时，我们常常会遇到需要重试的情况，比如网络请求失败、数据库连接错误等。为了提高代码的可靠性和稳定性，我们可以使用装饰器来实现函数的重试机制。

装饰器是一种Python语言特性，可以在函数执行前后加上额外的功能。通过使用装饰器，我们可以在不修改原函数代码的情况下，为函数添加新的功能。

下面是一个使用装饰器实现函数重试机制的例子：

import functools
import time

def retry(retries=3, delay=0):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            for i in range(retries):
                try:
                    result = func(*args, **kwargs)
                    return result
                except Exception as e:
                    print(f"Retry {i+1}/{retries}: {e}")
                    time.sleep(delay)
            raise Exception(f"Retry {retries} times, still failed.")
        return wrapper
    return decorator

上述代码定义了一个名为retry的装饰器函数，它接受两个参数retries和delay。retries表示最大重试次数，默认为3次；delay表示每次重试之间的延迟时间，默认为0秒。

装饰器函数内部定义了一个名为decorator的函数，它接受一个函数作为参数，并返回一个新的函数wrapper。

新的函数wrapper会在原函数的基础上加上重试的逻辑。它使用for循环进行多次重试，每次重试前会先执行原函数。如果原函数执行成功，则直接返回结果；如果原函数执行失败，则打印错误信息并等待一段时间后再次重试。如果重试次数达到最大次数后仍失败，抛出一个异常。

最后，在函数定义时使用装饰器即可实现重试机制。例如：

@retry(retries=5, delay=1)
def connect_to_database():
    # 使用数据库连接进行操作，这里简化为直接抛出异常
    raise Exception("Failed to connect to database.")

connect_to_database()

上述代码使用retry装饰器装饰了connect_to_database函数，并设置了重试次数为5次，重试之间的延迟为1秒。当我们调用connect_to_database函数时，如果出现异常，会自动进行重试。

通过使用装饰器，我们可以方便地为多个函数添加重试机制，提高代码的可靠性和稳定性。同时，装饰器还具有良好的可读性和复用性，可以减少重复代码的编写。