python3-core-api-client 1.2__py3-none-any.whl → 2.1.0__py3-none-any.whl

cyberfusion/CoreApiClient/connector.py

@@ -131,6 +131,7 @@ class CoreApiClient:
             status_code=requests_response.status_code,
             body=requests_response.text,
             headers=requests_response.headers,
+            requests_response=requests_response,
         )
 
         return local_response

cyberfusion/CoreApiClient/http.py

@@ -1,16 +1,25 @@
 import json
 from dataclasses import dataclass
 from http import HTTPStatus
-from typing import Any
+from typing import Any, TypeVar, Generic
 
 from requests.structures import CaseInsensitiveDict
 
+from cyberfusion.CoreApiClient.models import CoreApiModel
+
+from requests.models import Response as RequestsResponse
+
+
+ModelType = TypeVar("ModelType", bound=CoreApiModel)
+DtoType = TypeVar("DtoType", CoreApiModel, list[CoreApiModel])
+
 
 @dataclass
 class Response:
     status_code: int
     body: str
     headers: CaseInsensitiveDict
+    requests_response: RequestsResponse
 
     @property
     def failed(self) -> bool:
@@ -19,3 +28,23 @@ class Response:
     @property
     def json(self) -> Any:
         return json.loads(self.body)
+
+
+@dataclass
+class DtoResponse(Generic[DtoType], Response):
+    dto: DtoType
+
+    @classmethod
+    def from_response(cls, response: Response, model: type[ModelType]) -> "DtoResponse":
+        if isinstance(response.json, list):
+            dto = [model.parse_obj(object_) for object_ in response.json]
+        else:
+            dto = model.parse_obj(response.json)
+
+        return cls(
+            status_code=response.status_code,
+            body=response.body,
+            headers=response.headers,
+            requests_response=response.requests_response,
+            dto=dto,
+        )