documente_shared 0.1.81__tar.gz → 0.1.82__tar.gz

{documente_shared-0.1.81 → documente_shared-0.1.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.81
+Version: 0.1.82
 Summary: Shared utilities for Documente AI projects
 License: MIT
 Author: Tech

{documente_shared-0.1.81 → documente_shared-0.1.82}/documente_shared/domain/entities/processing_case.py

@@ -98,7 +98,7 @@ class ProcessingCase(object):
     def to_persist_dict(self) -> dict:
         persist_data = self.to_dict
         persist_data["items"] = [
-            item.to_persist_dict for item in self.items
+            item.to_dict for item in self.items
         ]
         return persist_data
 

{documente_shared-0.1.81 → documente_shared-0.1.82}/documente_shared/domain/entities/processing_case_item.py

@@ -106,7 +106,10 @@ class ProcessingCaseItem(object):
             'case_id': self.case_id,
             'digest': self.digest,
             'status': str(self.status),
-            'document': self.document.to_dict,
+            'document':(
+                self.document.to_dict
+                if self.document else None
+            ),
             'document_type': (
                 str(self.document_type)
                 if self.document_type else None

{documente_shared-0.1.81 → documente_shared-0.1.82}/documente_shared/infrastructure/repositories/http_processing_case.py

@@ -1,6 +1,8 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+from requests import Response
+
 from documente_shared.application.payloads import camel_to_snake
 from documente_shared.domain.entities.processing_case import ProcessingCase
 from documente_shared.domain.entities.processing_case_filters import ProcessingCaseFilters
@@ -16,9 +18,7 @@ class HttpProcessingCaseRepository(
     def find(self, uuid: str, include_items: bool = False) -> Optional[ProcessingCase]:
         response = self.session.get(f"{self.api_url}/v1/processing-cases/{uuid}/")
         if response.status_code == 200:
-            response_json = response.json()
-            instance_data = response_json.get('data', {})
-            return ProcessingCase.from_persist_dict(camel_to_snake(instance_data))
+            return self._build_processing_case(response)
         return None
 
     def persist(self, instance: ProcessingCase) -> ProcessingCase:
@@ -29,9 +29,7 @@ class HttpProcessingCaseRepository(
         if response.status_code not in [200, 201]:
             raise Exception(f'Error persisting processing case: {response.text}')
 
-        response_json = response.json()
-        instance_data = response_json.get('data', {})
-        return ProcessingCase.from_persist_dict(camel_to_snake(instance_data))
+        return self._build_processing_case(response)
 
     def remove(self, instance: ProcessingCase):
         self.session.delete(f"{self.api_url}/v1/processing-cases/{instance.uuid}/")
@@ -45,9 +43,14 @@ class HttpProcessingCaseRepository(
         )
         if response.status_code == 200:
             raw_response = response.json()
-            instaces_data = raw_response.get('data', [])
             return [
-                ProcessingCase.from_persist_dict(item_data)
-                for item_data in camel_to_snake(instaces_data)
+                ProcessingCase.from_persist_dict(camel_to_snake(item_data))
+                for item_data in raw_response.get('data', [])
             ]
-        return []
+        return []
+
+    @classmethod
+    def _build_processing_case(cls, response: Response) -> ProcessingCase:
+        response_json = response.json()
+        instance_data = response_json.get('data', {})
+        return ProcessingCase.from_persist_dict(camel_to_snake(instance_data))

{documente_shared-0.1.81 → documente_shared-0.1.82}/documente_shared/infrastructure/repositories/http_processing_case_item.py

@@ -1,6 +1,9 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+from requests import Response
+
+from documente_shared.application.payloads import camel_to_snake
 from documente_shared.domain.entities.processing_case_item import ProcessingCaseItem
 from documente_shared.domain.entities.processing_case_item_filters import ProcessingCaseItemFilters
 from documente_shared.domain.repositories.processing_case_item import ProcessingCaseItemRepository
@@ -12,41 +15,46 @@ class HttpProcessingCaseItemRepository(
     DocumenteClientMixin,
     ProcessingCaseItemRepository,
 ):
-
     def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
-        response = self.session.get(f"{self.api_url}/processing-case-items/{uuid}/")
+        response = self.session.get(f"{self.api_url}/v1/processing-case-items/{uuid}/")
         if response.status_code == 200:
-            return ProcessingCaseItem.from_persist_dict(response.json())
+            return self._build_processing_case_item(response)
         return None
 
     def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
-        response = self.session.get(f"{self.api_url}/processing-case-items/{digest}/")
+        response = self.session.get(f"{self.api_url}/v1/processing-case-items/{digest}/")
         if response.status_code == 200:
-            return ProcessingCaseItem.from_persist_dict(response.json())
+            return self._build_processing_case_item(response)
         return None
 
     def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
-        response = self.session.put(
-            url=f"{self.api_url}/processing-case-items/{instance.uuid}/",
+        response: Response = self.session.put(
+            url=f"{self.api_url}/v1/processing-case-items/{instance.uuid}/",
             json=instance.to_persist_dict,
         )
         if response.status_code in [200, 201]:
-            return ProcessingCaseItem.from_dict(response.json())
+            return self._build_processing_case_item(response)
         return instance
 
     def remove(self, instance: ProcessingCaseItem):
-        self.session.delete(f"{self.api_url}/processing-case-items/{instance.uuid}/")
+        self.session.delete(f"{self.api_url}/v1/processing-case-items/{instance.uuid}/")
 
     def filter(
         self,
         tenant_slug: str,
         filters: ProcessingCaseItemFilters,
     ) -> List[ProcessingCaseItem]:
-        response = self.session.get(f"{self.api_url}/processing-case-items/")
+        response = self.session.get(f"{self.api_url}/v1/processing-case-items/")
         if response.status_code == 200:
             raw_response = response.json()
             return [
-                ProcessingCaseItem.from_persist_dict(item)
-                for item in raw_response.get('data', [])
+                ProcessingCaseItem.from_dict(camel_to_snake(item_data))
+                for item_data in raw_response.get('data', [])
             ]
         return []
+
+    @classmethod
+    def _build_processing_case_item(cls, response: Response) -> ProcessingCaseItem:
+        response_json = response.json()
+        instance_data = response_json.get('data', {})
+        return ProcessingCaseItem.from_dict(camel_to_snake(instance_data))

{documente_shared-0.1.81 → documente_shared-0.1.82}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.81"
+version = "0.1.82"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"