How best to implement a Logging Adapter? #140

class LoggingHTTPAdapter(requests.adapters.HTTPAdapter):
    """Adapter to log request and response."""

    def send(self, request, *args, **kwargs):
        sensitive_params = conf.SENSITIVE_PARAMS

        # transaction ID to identify the logs.
        trans_id = util.generate_uuid()

        headers = util.filter_sensitive_information(
            request.headers, sensitive_params
        )

        log.info(f"<{trans_id}> Request: {request.method} {request.url}")
        log.info(f"<{trans_id}> Request headers: {headers}")
        log.info(f"<{trans_id}> Request body: {request.body}")

        resp = super().send(request, *args, **kwargs)

        headers = util.filter_sensitive_information(
            resp.headers, sensitive_params
        )

        log.info(f"<{trans_id}> Response: {resp.status_code} {resp.reason}")
        log.info(f"<{trans_id}> Response headers: {headers}")
        log.info(f"<{trans_id}> Response body: {resp.content}")

        # Store the transaction ID in the response object, so that it can be
        # added to the logs from test cases.
        resp.__dict__["transaction_id"] = trans_id

        return resp


def mount_adapter(session):
    # currently we only have TimeoutHTTPAdapter and LoggingHTTPAdapter,
    # when we add more in the future, we can use this function to mount
    # our adapters.
    adapters = [LoggingHTTPAdapter()]
    for adapter in adapters:
        session.mount("http://", adapter)
        session.mount("https://", adapter)