Tenacity 开源项目教程

Tenacity 开源项目教程

tenacity项目地址:https://gitcode.com/gh_mirrors/te/tenacity

项目介绍

Tenacity 是一个 Python 库，用于处理重试逻辑，提供了简单而强大的方式来处理网络请求、数据库操作等可能失败的操作。它基于 retrying 库，但提供了更多的功能和更好的性能。Tenacity 的主要特点包括：

• 支持多种重试策略（如固定间隔、指数退避等）
• 支持条件重试（如根据异常类型或返回值）
• 支持异步操作
• 简单易用的 API

项目快速启动

安装

首先，使用 pip 安装 Tenacity：

pip install tenacity

基本使用

以下是一个简单的示例，展示了如何使用 Tenacity 进行重试：

from tenacity import retry, wait_fixed

@retry(wait=wait_fixed(2))
def fetch_data():
    print("Trying to fetch data...")
    raise Exception("Failed to fetch data")

if __name__ == "__main__":
    fetch_data()

在这个示例中，fetch_data 函数会在每次失败后等待 2 秒，然后重试，直到成功或达到最大重试次数。

应用案例和最佳实践

网络请求重试

在处理网络请求时，重试逻辑尤为重要。以下是一个使用 Tenacity 处理 HTTP 请求的示例：

import requests
from tenacity import retry, wait_exponential, stop_after_attempt

@retry(wait=wait_exponential(multiplier=1, max=10), stop=stop_after_attempt(5))
def make_request():
    response = requests.get("https://api.example.com/data")
    response.raise_for_status()
    return response.json()

if __name__ == "__main__":
    data = make_request()
    print(data)

在这个示例中，make_request 函数会在每次失败后等待指数增长的时间（最多 10 秒），并且最多重试 5 次。

数据库操作重试

在处理数据库操作时，重试逻辑同样重要。以下是一个使用 Tenacity 处理数据库连接的示例：

import psycopg2
from tenacity import retry, wait_fixed, stop_after_delay

@retry(wait=wait_fixed(3), stop=stop_after_delay(30))
def connect_to_db():
    conn = psycopg2.connect(
        dbname="mydb",
        user="myuser",
        password="mypassword",
        host="localhost",
        port="5432"
    )
    return conn

if __name__ == "__main__":
    conn = connect_to_db()
    print("Connected to database")

在这个示例中，connect_to_db 函数会在每次失败后等待 3 秒，并且最多重试 30 秒。

典型生态项目

Tenacity 可以与其他 Python 库和框架结合使用，以提供更强大的功能。以下是一些典型的生态项目：

• Requests: 用于处理 HTTP 请求的库，可以与 Tenacity 结合使用以处理网络请求的重试逻辑。
• SQLAlchemy: 用于处理数据库操作的库，可以与 Tenacity 结合使用以处理数据库连接的重试逻辑。
• Celery: 用于处理异步任务的库，可以与 Tenacity 结合使用以处理任务的重试逻辑。

通过结合这些生态项目，Tenacity 可以提供更全面的重试解决方案，确保应用程序的稳定性和可靠性。

tenacity项目地址:https://gitcode.com/gh_mirrors/te/tenacity