itkdb

def delete_from_eos(self, response, **_) -> None:
    """
    requests response hook function to delete a file from eos.
    """
    try:
        response.raise_for_status()
    except HTTPError:
        log.warning("Something went wrong with deleting the attachment.")
        return response

    data = response.json()

    # do nothing if it's not an EOS-type attachment
    # or if betamax is being used (no need to run the cURL for EOS)
    if (
        data["attachment"]["type"] != "eos"
        or response.connection.__class__.__name__ == "BetamaxAdapter"
    ):
        return None

    if "token" not in data:
        log.warning(
            "It seems there is no token, so we are not deleting this from EOS."
        )
        return None

    log.info(
        "It looks like you're deleting an attachment from ITk PD that is stored on EOS, I will try to delete it from EOS for you."
    )

    response.eos_response = eos.delete(data["token"], data["attachment"]["url"])
    return None

def get(self, url, **kwargs):
    is_cern_url = ".cern.ch" in urlparse(url).netloc
    # is_binary_data = "uu-app-binarystore/getBinaryData" in url
    if is_cern_url:
        log.info(
            "Identified a cern.ch request, will attach CERN SSL chain to request by overriding `verify`."
        )
        kwargs["verify"] = itkdb_data / "CERN_chain.pem"

    # getBinaryData does not handle chunked requests
    # if is_cern_url or is_binary_data:
    if is_cern_url:
        log.info(
            "Identified a request that potentially downloads larger amounts of data, will execute chunked requests (stream=True)."
        )
        kwargs["stream"] = True
        headers = kwargs.get("headers", {})
        headers["transfer-encoding"] = "chunked"
        kwargs["headers"] = headers
    return super().get(url, **kwargs)

def prepare_request(self, request):
    request.url = self._normalize_url(request.url)
    self._request_handler(request)
    return super().prepare_request(request)

def request(self, method, url, *args, **kwargs):
    self.limit = kwargs.pop("limit", -1)

    response = super(Session, self).request(method, url, *args, **kwargs)
    return self._response_handler(response)

def upload_to_eos(self, response, eos_file_details=None, **_) -> None:
    """
    requests response hook function to upload a file to eos.
    """
    log.info("I was able to get a token to upload to EOS. Let me upload.")
    try:
        response.raise_for_status()
    except HTTPError:
        log.warning("Something went wrong with uploading to EOS.")
        return response

    # do nothing if betamax is being used (no need to run the cURL for EOS)
    if response.connection.__class__.__name__ == "BetamaxAdapter":
        return None

    token_request = response.json()

    log.info(token_request)

    response.eos_response = eos.put(
        token_request["token"], token_request["url"], eos_file_details
    )
    return None