5222093b54
* support custom error messaging for error response + retryable errors * remove changed backoff i was using for testing * refactor filter to construct response status internally * pr feedback * bump version and update changelog
42 lines
1.3 KiB
Python
42 lines
1.3 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: str = ""):
|
|
error_message = (
|
|
error_message or 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, error_message: str = ""):
|
|
"""
|
|
: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, error_message=error_message)
|
|
|
|
|
|
class DefaultBackoffException(BaseBackoffException):
|
|
pass
|