documente_shared 0.1.96__tar.gz → 0.1.98__tar.gz

{documente_shared-0.1.96 → documente_shared-0.1.98}/PKG-INFO

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

documente_shared-0.1.96/documente_shared/domain/entities/in_memory_result.py → documente_shared-0.1.98/documente_shared/domain/entities/in_memory_document.py

@@ -1,10 +1,12 @@
-from dataclasses import dataclass
+import base64
 from typing import Optional
+from dataclasses import dataclass
 
 from documente_shared.application.files import (
     remove_slash_from_path,
     get_filename_from_path,
 )
+from documente_shared.domain.exceptions import InMemoryDocumentContentError
 
 
 @dataclass
@@ -48,6 +50,18 @@ class InMemoryDocument(object):
 
     @classmethod
     def from_dict(cls, data: dict):
+        has_bytes_content = data.get('file_bytes') and isinstance(data['file_bytes'], bytes)
+        has_base64_content = data.get('file_base64') and isinstance(data['file_base64'], str)
+
+        if not has_bytes_content and not has_base64_content:
+            raise InMemoryDocumentContentError
+
+        if has_bytes_content and not data.get('file_base64'):
+            data['file_base64'] = base64.b64encode(data.get('file_bytes')).decode("utf-8")
+
+        if has_base64_content and not data.get('file_bytes'):
+            data['file_bytes'] = base64.b64decode(data.get('file_base64'))
+
         return cls(
             file_path=data.get('file_path'),
             file_bytes=data.get('file_bytes'),

{documente_shared-0.1.96 → documente_shared-0.1.98}/documente_shared/domain/entities/processing_case.py

@@ -88,9 +88,10 @@ class ProcessingCase(object):
     def to_dict(self) -> dict:
         return {
             'uuid': self.uuid,
-            'label': self.name,
+            'tenant_slug': self.tenant_slug,
+            'name': self.name,
             'status': str(self.status),
-            'category': (
+            'case_type': (
                 str(self.case_type)
                 if self.case_type else None
             ),

{documente_shared-0.1.96 → documente_shared-0.1.98}/documente_shared/domain/entities/processing_case_item.py

@@ -5,7 +5,7 @@ from typing import Optional, List
 
 from documente_shared.application.time_utils import get_datetime_from_data
 from documente_shared.domain.constants import la_paz_tz
-from documente_shared.domain.entities.in_memory_result import InMemoryDocument
+from documente_shared.domain.entities.in_memory_document import InMemoryDocument
 from documente_shared.domain.enums.common import ProcessingStatus, ProcessingSource
 from documente_shared.domain.enums.processing_case import ProcessingDocumentType
 

documente_shared-0.1.98/documente_shared/domain/exceptions.py

@@ -0,0 +1,5 @@
+
+
+
+class InMemoryDocumentContentError(Exception):
+    pass

documente_shared-0.1.98/documente_shared/infrastructure/repositories/mem_document.py

@@ -0,0 +1,37 @@
+from dataclasses import dataclass
+from typing import List, Optional
+
+from documente_shared.domain.entities.document import DocumentProcessing
+from documente_shared.domain.enums.document import DocumentProcessingStatus
+from documente_shared.domain.repositories.document import DocumentProcessingRepository
+
+
+@dataclass
+class MemoryDocumentProcessingRepository(DocumentProcessingRepository):
+    collection: dict[str, DocumentProcessing] = None
+
+    def __post_init__(self):
+        self.collection = self.collection or {}
+
+    def find(self, digest: str) -> Optional[DocumentProcessing]:
+        if digest in self.collection:
+            return self.collection[digest]
+        return None
+
+
+    def persist(self, instance: DocumentProcessing) -> DocumentProcessing:
+        self.collection[instance.digest] = instance
+        return instance
+
+    def remove(self, instance: DocumentProcessing):
+        if instance.digest in self.collection:
+            del self.collection[instance.digest]
+        return None
+
+    def filter(self, statuses: List[DocumentProcessingStatus]) -> List[DocumentProcessing]:
+        items = []
+        for status in statuses:
+            items.extend(
+                [item for item in self.collection.values() if item.status == status]
+            )
+        return items

documente_shared-0.1.98/documente_shared/infrastructure/repositories/mem_processing_case.py

@@ -0,0 +1,36 @@
+from dataclasses import dataclass
+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.repositories.processing_case import ProcessingCaseRepository
+
+
+@dataclass
+class MemoryProcessingCaseRepository(ProcessingCaseRepository):
+    collection: dict[str, ProcessingCase] = None
+
+    def __post_init__(self):
+        self.collection = self.collection or {}
+
+    def find(
+        self,
+        uuid: str,
+        include_items: bool = False,
+    ) -> Optional[ProcessingCase]:
+        if uuid in self.collection:
+            return self.collection[uuid]
+        return None
+
+    def persist(self, instance: ProcessingCase) -> ProcessingCase:
+        self.collection[instance.uuid] = instance
+        return instance
+
+    def remove(self, instance: ProcessingCase):
+        if instance.uuid in self.collection:
+            del self.collection[instance.uuid]
+        return None
+
+    def filter(self, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
+        return []

{documente_shared-0.1.96 → documente_shared-0.1.98}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.96"
+version = "0.1.98"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"
@@ -15,9 +15,10 @@ pytz = "^2025.2"
 requests = "^2.32.3"
 loguru = "^0.7.3"
 
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
 pytest = "^8.3.4"
 ipdb = "^0.13.13"
+expects = "^0.9.0"
 
 
 [tool.pytest.ini_options]