documente_shared 0.1.113__tar.gz → 0.1.115__tar.gz

{documente_shared-0.1.113 → documente_shared-0.1.115}/PKG-INFO

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

{documente_shared-0.1.113 → documente_shared-0.1.115}/documente_shared/application/query_params.py

@@ -14,6 +14,10 @@ def camelize_string(value: str) -> str:
     return value[:1].lower() + value[1:]
 
 
+def parse_bool(input_value: bool) -> str:
+    return str(input_value).lower()
+
+
 @dataclass
 class QueryParams(object):
     params: Union[dict]

{documente_shared-0.1.113 → documente_shared-0.1.115}/documente_shared/domain/entities/processing_case_item.py

@@ -17,7 +17,7 @@ class ProcessingCaseItem(object):
     case_id: str
     digest: str
     status: ProcessingStatus
-    document: InMemoryDocument
+    document: Optional[InMemoryDocument] = None
     document_type: Optional[ProcessingDocumentType] = None
     uploaded_from: Optional[ProcessingSource] = None
     processed_csv: Optional[InMemoryDocument] = None
@@ -64,9 +64,9 @@ class ProcessingCaseItem(object):
         self.started_at = datetime.now(tz=timezone)
 
     def failed(
-            self,
-            error_message: Optional[str] = None,
-            timezone: tzinfo = la_paz_tz,
+        self,
+        error_message: Optional[str] = None,
+        timezone: tzinfo = la_paz_tz,
     ):
         self.status = ProcessingStatus.FAILED
         self.failed_at = datetime.now(tz=timezone)

{documente_shared-0.1.113 → documente_shared-0.1.115}/documente_shared/domain/repositories/document.py

@@ -8,11 +8,11 @@ from documente_shared.domain.enums.document import DocumentProcessingStatus
 class DocumentProcessingRepository(ABC):
 
     @abstractmethod
-    def find(self, digest: str) -> Optional[DocumentProcessing]:
+    def find(self, digest: str, read_bytes: bool = False) -> Optional[DocumentProcessing]:
         raise NotImplementedError
 
     @abstractmethod
-    def persist(self, instance: DocumentProcessing) -> DocumentProcessing:
+    def persist(self, instance: DocumentProcessing, read_bytes: bool = False) -> DocumentProcessing:
         raise NotImplementedError
 
     @abstractmethod

{documente_shared-0.1.113 → documente_shared-0.1.115}/documente_shared/domain/repositories/processing_case_item.py

@@ -8,15 +8,27 @@ from documente_shared.domain.entities.processing_case_item_filters import Proces
 class ProcessingCaseItemRepository(ABC):
 
     @abstractmethod
-    def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
+    def find(
+        self,
+        uuid: str,
+        read_bytes: bool = False,
+    ) -> Optional[ProcessingCaseItem]:
         raise NotImplementedError
 
     @abstractmethod
-    def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
+    def find_by_digest(
+        self,
+        digest: str,
+        read_bytes: bool = False,
+    ) -> Optional[ProcessingCaseItem]:
         raise NotImplementedError
 
     @abstractmethod
-    def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
+    def persist(
+        self,
+        instance: ProcessingCaseItem,
+        read_bytes: bool = False,
+    ) -> ProcessingCaseItem:
         raise NotImplementedError
 
     @abstractmethod

{documente_shared-0.1.113 → documente_shared-0.1.115}/documente_shared/infrastructure/repositories/dynamo_processing_case_item.py

@@ -13,19 +13,19 @@ class DynamoProcessingCaseItemRepository(
     DynamoDBTable,
     ProcessingCaseItemRepository,
 ):
-    def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
+    def find(self, uuid: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
         item = self.get(key={'digest': uuid})
         if item:
             return ProcessingCaseItem.from_dict(item)
         return None
 
-    def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
+    def find_by_digest(self, digest: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
         item = self.get(key={'digest': digest})
         if item:
             return ProcessingCaseItem.from_dict(item)
         return None
 
-    def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
+    def persist(self, instance: ProcessingCaseItem, read_bytes: bool = False) -> ProcessingCaseItem:
         self.put(instance.to_simple_dict)
         return instance
 

{documente_shared-0.1.113 → documente_shared-0.1.115}/documente_shared/infrastructure/repositories/http_processing_case_item.py

@@ -5,6 +5,7 @@ from loguru import logger
 from requests import Response
 
 from documente_shared.application.payloads import camel_to_snake
+from documente_shared.application.query_params import parse_bool
 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
@@ -16,22 +17,26 @@ class HttpProcessingCaseItemRepository(
     DocumenteClientMixin,
     ProcessingCaseItemRepository,
 ):
-    def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
-        response = self.session.get(f"{self.api_url}/v1/processing-case-items/{uuid}/")
+    def find(self, uuid: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
+        response = self.session.get(
+            url=f"{self.api_url}/v1/processing-case-items/{uuid}/?read_bytes={parse_bool(read_bytes)}",
+        )
         if response.status_code not in [200, 201]:
             return None
         return self._build_processing_case_item(response)
 
-    def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
-        response = self.session.get(f"{self.api_url}/v1/processing-case-items/{digest}/")
+    def find_by_digest(self, digest: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
+        response = self.session.get(
+            url=f"{self.api_url}/v1/processing-case-items/{digest}/?read_bytes={parse_bool(read_bytes)}",
+        )
         if response.status_code not in [200, 201]:
             return None
         return self._build_processing_case_item(response)
 
-    def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
+    def persist(self, instance: ProcessingCaseItem, read_bytes: bool = False) -> ProcessingCaseItem:
         logger.info(f"PERSISTING_PROCESSING_CASE_ITEM: data={instance.to_simple_dict}")
         response: Response = self.session.put(
-            url=f"{self.api_url}/v1/processing-case-items/{instance.uuid}/",
+            url=f"{self.api_url}/v1/processing-case-items/{instance.uuid}/?read_bytes={parse_bool(read_bytes)}",
             json=instance.to_persist_dict,
         )
         if response.status_code not in [200, 201]:

{documente_shared-0.1.113 → documente_shared-0.1.115}/documente_shared/infrastructure/repositories/mem_processing_case_item.py

@@ -17,20 +17,28 @@ class MemoryProcessingCaseItemRepository(ProcessingCaseItemRepository):
     def find(
         self,
         uuid: str,
-        include_items: bool = False,
+        read_items: bool = False,
     ) -> Optional[ProcessingCaseItem]:
         if uuid in self.collection:
             return self.collection[uuid]
         return None
 
-    def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
+    def find_by_digest(
+        self,
+        digest: str,
+        read_bytes: bool = False
+    ) -> Optional[ProcessingCaseItem]:
         for item in self.collection.values():
             if item.digest == digest:
                 return item
         return None
 
 
-    def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
+    def persist(
+        self,
+        instance: ProcessingCaseItem,
+        read_bytes: bool = False,
+    ) -> ProcessingCaseItem:
         self.collection[instance.uuid] = instance
         return instance
 

{documente_shared-0.1.113 → documente_shared-0.1.115}/pyproject.toml

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