40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
#
|
|
# Copyright (c) 2022 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
|
|
from typing import Union
|
|
|
|
import requests
|
|
|
|
|
|
class BaseBackoffException(requests.exceptions.HTTPError):
|
|
def __init__(self, request: requests.PreparedRequest, response: requests.Response):
|
|
error_message = f"Request URL: {request.url}, Response Code: {response.status_code}, Response Text: {response.text}"
|
|
super().__init__(error_message, request=request, response=response)
|
|
|
|
|
|
class RequestBodyException(Exception):
|
|
"""
|
|
Raised when there are issues in configuring a request body
|
|
"""
|
|
|
|
|
|
class UserDefinedBackoffException(BaseBackoffException):
|
|
"""
|
|
An exception that exposes how long it attempted to backoff
|
|
"""
|
|
|
|
def __init__(self, backoff: Union[int, float], request: requests.PreparedRequest, response: requests.Response):
|
|
"""
|
|
:param backoff: how long to backoff in seconds
|
|
:param request: the request that triggered this backoff exception
|
|
:param response: the response that triggered the backoff exception
|
|
"""
|
|
self.backoff = backoff
|
|
super().__init__(request=request, response=response)
|
|
|
|
|
|
class DefaultBackoffException(BaseBackoffException):
|
|
pass
|