documente_shared 0.1.125__tar.gz → 0.1.127__tar.gz

{documente_shared-0.1.125 → documente_shared-0.1.127}/PKG-INFO

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

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/domain/entities/in_memory_document.py

@@ -3,7 +3,6 @@ from typing import Optional
 from dataclasses import dataclass
 
 from documente_shared.application.files import get_filename_from_path, remove_extension, remove_slash_from_path
-from documente_shared.domain.exceptions import InMemoryDocumentContentError
 
 
 @dataclass

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/domain/enums/common.py

@@ -117,3 +117,17 @@ class TaskResultStatus(BaseEnum):
     SUCCESS = "SUCCESS"
     IN_PROGRESS = "IN_PROGRESS"
     FAILURE = "FAILURE"
+
+
+class DocumentViewFormat(BaseEnum):
+    PUBLIC_URL = 'PUBLIC_URL'
+    STORAGE_KEY = 'STORAGE_KEY'
+
+    @property
+    def is_public_url(self):
+        return self == DocumentViewFormat.PUBLIC_URL
+
+    @property
+    def is_object_storage(self):
+        return self == DocumentViewFormat.STORAGE_KEY
+

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/domain/repositories/processing_case.py

@@ -3,6 +3,7 @@ from typing import Optional, List
 
 from documente_shared.domain.entities.processing_case import ProcessingCase
 from documente_shared.domain.entities.processing_case_filters import ProcessingCaseFilters
+from documente_shared.domain.enums.common import DocumentViewFormat
 
 
 class ProcessingCaseRepository(ABC):
@@ -13,6 +14,7 @@ class ProcessingCaseRepository(ABC):
         uuid: str,
         include_items: bool = False,
         include_items_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> Optional[ProcessingCase]:
         raise NotImplementedError
 
@@ -21,6 +23,7 @@ class ProcessingCaseRepository(ABC):
         self,
         instance: ProcessingCase,
         persist_items: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> ProcessingCase:
         raise NotImplementedError
 

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/domain/repositories/processing_case_item.py

@@ -3,6 +3,7 @@ from typing import Optional, List
 
 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.enums.common import DocumentViewFormat
 
 
 class ProcessingCaseItemRepository(ABC):
@@ -12,6 +13,7 @@ class ProcessingCaseItemRepository(ABC):
         self,
         uuid: str,
         read_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> Optional[ProcessingCaseItem]:
         raise NotImplementedError
 
@@ -20,6 +22,7 @@ class ProcessingCaseItemRepository(ABC):
         self,
         digest: str,
         read_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> Optional[ProcessingCaseItem]:
         raise NotImplementedError
 
@@ -29,6 +32,7 @@ class ProcessingCaseItemRepository(ABC):
         instance: ProcessingCaseItem,
         read_bytes: bool = False,
         persist_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> ProcessingCaseItem:
         raise NotImplementedError
 
@@ -40,5 +44,6 @@ class ProcessingCaseItemRepository(ABC):
     def filter(
         self,
         filters: ProcessingCaseItemFilters,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> List[ProcessingCaseItem]:
         raise NotImplementedError

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/infrastructure/repositories/dynamo_processing_case.py

@@ -4,6 +4,7 @@ from boto3.dynamodb.conditions import Key
 
 from documente_shared.domain.entities.processing_case import ProcessingCase
 from documente_shared.domain.entities.processing_case_filters import ProcessingCaseFilters
+from documente_shared.domain.enums.common import DocumentViewFormat
 from documente_shared.domain.repositories.processing_case import ProcessingCaseRepository
 
 from documente_shared.infrastructure.dynamo_table import DynamoDBTable
@@ -18,6 +19,7 @@ class DynamoProcessingCaseRepository(
         uuid: str,
         include_items: bool = False,
         include_items_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> Optional[ProcessingCase]:
         item = self.get(key={'uuid': uuid})
         if item:
@@ -28,6 +30,7 @@ class DynamoProcessingCaseRepository(
         self,
         instance: ProcessingCase,
         persist_items: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> ProcessingCase:
         self.put(instance.to_persist_dict)
         return instance

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/infrastructure/repositories/dynamo_processing_case_item.py

@@ -4,6 +4,7 @@ from boto3.dynamodb.conditions import Key
 
 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.enums.common import DocumentViewFormat
 from documente_shared.domain.repositories.processing_case_item import ProcessingCaseItemRepository
 
 from documente_shared.infrastructure.dynamo_table import DynamoDBTable
@@ -13,13 +14,23 @@ class DynamoProcessingCaseItemRepository(
     DynamoDBTable,
     ProcessingCaseItemRepository,
 ):
-    def find(self, uuid: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
+    def find(
+        self,
+        uuid: str,
+        read_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
+    ) -> Optional[ProcessingCaseItem]:
         item = self.get(key={'digest': uuid})
         if item:
             return ProcessingCaseItem.from_dict(item)
         return None
 
-    def find_by_digest(self, digest: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
+    def find_by_digest(
+        self,
+        digest: str,
+        read_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
+    ) -> Optional[ProcessingCaseItem]:
         item = self.get(key={'digest': digest})
         if item:
             return ProcessingCaseItem.from_dict(item)
@@ -30,6 +41,7 @@ class DynamoProcessingCaseItemRepository(
         instance: ProcessingCaseItem,
         read_bytes: bool = False,
         persist_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> ProcessingCaseItem:
         self.put(instance.to_simple_dict)
         return instance
@@ -40,6 +52,7 @@ class DynamoProcessingCaseItemRepository(
     def filter(
         self,
         filters: ProcessingCaseItemFilters,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> List[ProcessingCaseItem]:
         items = []
 

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/infrastructure/repositories/http_processing_case.py

@@ -1,5 +1,4 @@
 from dataclasses import dataclass
-from sqlite3.dbapi2 import paramstyle
 from typing import List, Optional
 
 from loguru import logger
@@ -9,6 +8,7 @@ from documente_shared.application.payloads import camel_to_snake
 from documente_shared.application.query_params import parse_bool_to_str
 from documente_shared.domain.entities.processing_case import ProcessingCase
 from documente_shared.domain.entities.processing_case_filters import ProcessingCaseFilters
+from documente_shared.domain.enums.common import DocumentViewFormat
 from documente_shared.domain.repositories.processing_case import ProcessingCaseRepository
 from documente_shared.infrastructure.documente_client import DocumenteClientMixin
 
@@ -23,8 +23,15 @@ class HttpProcessingCaseRepository(
         uuid: str,
         include_items: bool = False,
         include_items_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> Optional[ProcessingCase]:
-        response = self.session.get(f"{self.api_url}/v1/processing-cases/{uuid}/")
+        params = {
+            "view_format": str(view_format),
+        }
+        response = self.session.get(
+            url=f"{self.api_url}/v1/processing-cases/{uuid}/",
+            params=params,
+        )
         if response.status_code not in [200, 201]:
             return None
         return self._build_processing_case(response)
@@ -33,10 +40,12 @@ class HttpProcessingCaseRepository(
         self,
         instance: ProcessingCase,
         persist_items: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> ProcessingCase:
         logger.info(f"PERSISTING_PROCESSING_CASE: data={instance.to_queue_dict}")
         params = {
             "persist_items": parse_bool_to_str(persist_items),
+            "view_format": str(view_format),
         }
         response = self.session.put(
             url=f"{self.api_url}/v1/processing-cases/{instance.uuid}/",

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/infrastructure/repositories/http_processing_case_item.py

@@ -8,6 +8,7 @@ from documente_shared.application.payloads import camel_to_snake
 from documente_shared.application.query_params import parse_bool_to_str
 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.enums.common import DocumentViewFormat
 from documente_shared.domain.repositories.processing_case_item import ProcessingCaseItemRepository
 from documente_shared.infrastructure.documente_client import DocumenteClientMixin
 
@@ -17,7 +18,12 @@ class HttpProcessingCaseItemRepository(
     DocumenteClientMixin,
     ProcessingCaseItemRepository,
 ):
-    def find(self, uuid: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
+    def find(
+        self,
+        uuid: str,
+        read_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
+    ) -> Optional[ProcessingCaseItem]:
         response = self.session.get(
             url=f"{self.api_url}/v1/processing-case-items/{uuid}/?read_bytes={parse_bool_to_str(read_bytes)}",
         )
@@ -25,9 +31,19 @@ class HttpProcessingCaseItemRepository(
             return None
         return self._build_processing_case_item(response)
 
-    def find_by_digest(self, digest: str, read_bytes: bool = False) -> Optional[ProcessingCaseItem]:
+    def find_by_digest(
+        self,
+        digest: str,
+        read_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
+    ) -> Optional[ProcessingCaseItem]:
+        params = {
+            "read_bytes": parse_bool_to_str(read_bytes),
+            "view_format": str(view_format),
+        }
         response = self.session.get(
-            url=f"{self.api_url}/v1/processing-case-items/{digest}/?read_bytes={parse_bool_to_str(read_bytes)}",
+            url=f"{self.api_url}/v1/processing-case-items/{digest}/",
+            params=params,
         )
         if response.status_code not in [200, 201]:
             return None
@@ -38,10 +54,12 @@ class HttpProcessingCaseItemRepository(
         instance: ProcessingCaseItem,
         read_bytes: bool = False,
         persist_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> ProcessingCaseItem:
         logger.info(f"PERSISTING_PROCESSING_CASE_ITEM: data={instance.to_queue_dict}")
         params = {
             "read_bytes": parse_bool_to_str(read_bytes),
+            "view_format": str(view_format),
         }
         response: Response = self.session.put(
             url=f"{self.api_url}/v1/processing-case-items/{instance.uuid}/",
@@ -62,8 +80,15 @@ class HttpProcessingCaseItemRepository(
     def filter(
         self,
         filters: ProcessingCaseItemFilters,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> List[ProcessingCaseItem]:
-        response = self.session.get(f"{self.api_url}/v1/processing-case-items/")
+        params = {
+            "view_format": str(view_format),
+        }
+        response = self.session.get(
+            url=f"{self.api_url}/v1/processing-case-items/",
+            params=params,
+        )
         if response.status_code not in [200, 201]:
             return []
         raw_response = response.json()

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/infrastructure/repositories/mem_document.py

@@ -2,6 +2,7 @@ from dataclasses import dataclass
 from typing import List, Optional
 
 from documente_shared.domain.entities.document import DocumentProcessing
+from documente_shared.domain.enums.common import DocumentViewFormat
 from documente_shared.domain.enums.document import DocumentProcessingStatus
 from documente_shared.domain.repositories.document import DocumentProcessingRepository
 
@@ -13,13 +14,21 @@ class MemoryDocumentProcessingRepository(DocumentProcessingRepository):
     def __post_init__(self):
         self.collection = self.collection or {}
 
-    def find(self, digest: str) -> Optional[DocumentProcessing]:
+    def find(
+        self,
+        digest: str,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
+    ) -> Optional[DocumentProcessing]:
         if digest in self.collection:
             return self.collection[digest]
         return None
 
 
-    def persist(self, instance: DocumentProcessing) -> DocumentProcessing:
+    def persist(
+        self,
+        instance: DocumentProcessing,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
+    ) -> DocumentProcessing:
         self.collection[instance.digest] = instance
         return instance
 

{documente_shared-0.1.125 → documente_shared-0.1.127}/documente_shared/infrastructure/repositories/mem_processing_case.py

@@ -3,6 +3,7 @@ from typing import List, Optional
 
 from documente_shared.domain.entities.processing_case import ProcessingCase
 from documente_shared.domain.entities.processing_case_filters import ProcessingCaseFilters
+from documente_shared.domain.enums.common import DocumentViewFormat
 
 from documente_shared.domain.repositories.processing_case import ProcessingCaseRepository
 
@@ -19,6 +20,7 @@ class MemoryProcessingCaseRepository(ProcessingCaseRepository):
         uuid: str,
         include_items: bool = False,
         include_items_bytes: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> Optional[ProcessingCase]:
         if uuid in self.collection:
             return self.collection[uuid]
@@ -28,6 +30,7 @@ class MemoryProcessingCaseRepository(ProcessingCaseRepository):
         self,
         instance: ProcessingCase,
         persist_items: bool = False,
+        view_format: DocumentViewFormat = DocumentViewFormat.PUBLIC_URL,
     ) -> ProcessingCase:
         self.collection[instance.uuid] = instance
         return instance

{documente_shared-0.1.125 → documente_shared-0.1.127}/pyproject.toml

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