From ec5c385286c6da8190d00b0f075a82bf1970f845 Mon Sep 17 00:00:00 2001 From: pyoor Date: Mon, 6 Oct 2025 11:50:08 -0400 Subject: [PATCH] feat: automatically retry failed requests --- src/fuzzfetch/download.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/fuzzfetch/download.py b/src/fuzzfetch/download.py index ed111288..086fed1a 100644 --- a/src/fuzzfetch/download.py +++ b/src/fuzzfetch/download.py @@ -10,14 +10,25 @@ from typing import TYPE_CHECKING from requests import Response, Session +from requests.adapters import HTTPAdapter from requests.exceptions import RequestException +from urllib3 import Retry from .errors import FetcherException if TYPE_CHECKING: from .path import PathArg +HTTP_ADAPTER = HTTPAdapter( + max_retries=Retry( + total=3, + backoff_factor=1, + status_forcelist=[500, 502, 503, 504], + ) +) HTTP_SESSION = Session() +HTTP_SESSION.mount("https://", HTTP_ADAPTER) + LOG = getLogger("fuzzfetch")