Source code for django_rest_client.api_response

import requests
import json

from requests.models import CaseInsensitiveDict


class APIResponse(object):
    response: requests.Response
    url: str
    code: int
    headers: CaseInsensitiveDict
    data: dict
    file_name: str
    raw: bytes

    def __init__(self, response: requests.Response) -> None:
        self._response = response
        self.url = response.url
        self.code = response.status_code
        self.headers = response.headers
        try:
            self._handle_json_response()
        except json.JSONDecodeError:
            self._handle_generic_response()

    def _handle_json_response(self) -> None:
        self.data = self._response.json()

    def _handle_generic_response(self) -> None:
        content_disposition = self._response.headers.get("Content-Disposition", "")
        _split_names = content_disposition.split(r"attachment; filename=")
        if len(_split_names) == 2:
            # case: file attachment
            self.file_name = _split_names[1]
            self.raw = self._response.content

    def __repr__(self) -> str:
        return f'APIResponse("{self.url}", {self.code})'