Skip to content

Commit

Permalink
Merge pull request #6 from ImageMarkup/bigger-backoff
Browse files Browse the repository at this point in the history
bigger backoff
  • Loading branch information
danlamanna authored Jun 7, 2022
2 parents b245c61 + cfb72d1 commit d3cb62f
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions isic_cli/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,15 +2,29 @@
import time
from typing import Optional

from requests.packages.urllib3.util.retry import Retry
from retryable_requests import RetryableSession

logger = logging.getLogger('isic_cli')


# The same as retryable-requests DEFAULT_RETRY_STRATEGY with an
# increased backoff factor.
ISIC_RETRY_STRATEGY = Retry(
total=5,
status_forcelist=[429, 500, 502, 503, 504],
backoff_factor=5,
redirect=False,
raise_on_status=False,
)


class IsicCliSession(RetryableSession):
def __init__(self, *args, **kwargs) -> None:
from isic_cli.utils.version import get_version

kwargs.setdefault('retry_strategy', ISIC_RETRY_STRATEGY)

super().__init__(*args, **kwargs)

self.headers.update(
Expand Down

0 comments on commit d3cb62f

Please sign in to comment.