integration/helpers/request_utils.py

# Utils for requests

# Relevant headers that should be captured for debugging
AMAZON_HEADERS = [
    "x-amzn-requestid",
    "x-amz-apigw-id",
    "x-amz-cf-id",
    "x-amzn-errortype",
    "apigw-requestid",
]


class RequestUtils:
    def __init__(self, response):
        self.response = response
        self.headers = self._normalize_response_headers()

    def get_amazon_headers(self):
        """
        Get a list of relevant amazon headers that could be useful for debugging
        """
        amazon_headers = {}
        for header, header_val in self.headers.items():
            if header in AMAZON_HEADERS:
                amazon_headers[header] = header_val
        return amazon_headers

    def _normalize_response_headers(self):
        """
        API gateway can return headers with letters in different cases i.e. x-amzn-requestid or x-amzn-requestId
        We make them all lowercase here to more easily match them up
        """
        if self.response is None or not self.response.headers:
            # Need to check for response is None here since the __bool__ method checks 200 <= status < 400
            return {}

        return dict((k.lower(), v) for k, v in self.response.headers.items())