documente_shared 0.1.96__py3-none-any.whl → 0.1.98__py3-none-any.whl

documente_shared/domain/entities/{in_memory_result.py → 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/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/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/domain/exceptions.py

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

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/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.dist-info → documente_shared-0.1.98.dist-info}/METADATA

@@ -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.dist-info → documente_shared-0.1.98.dist-info}/RECORD

@@ -14,10 +14,10 @@ documente_shared/domain/constants.py,sha256=NG5BGaXBr_FnzudjTRPxpDpyiSDdaB_PLCdl
 documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/entities/document.py,sha256=xGZdrqqBkbvDbgoxUvIck6FplXjwLcZzFkcEsuh80u8,12819
 documente_shared/domain/entities/document_metadata.py,sha256=ygyFIC5qwxlm8DUM5kvVFny9zJfPQS8vNLM2br5XsQ8,2353
-documente_shared/domain/entities/in_memory_result.py,sha256=0sLNUrovKFQx4M-E9e4DrAiVgch2i4AKA-9BQBRaeI8,1482
-documente_shared/domain/entities/processing_case.py,sha256=ocPSvy0owiJq0cWnSHLIlY1UgE6Qz3ys0NwF6IxcMI0,5595
+documente_shared/domain/entities/in_memory_document.py,sha256=mOhylBu1k1Ke7G2NO0UHH7kCVp-H2PTa_4QZCNgUaAA,2161
+documente_shared/domain/entities/processing_case.py,sha256=2wD256_ra5szb8LVmGsauTsFTbkAqT1srOv5zuZQXgA,5640
 documente_shared/domain/entities/processing_case_filters.py,sha256=harKyu7QEuL1bI_Z8_UxkVCMo5r9vHeNHyi_Ja07vjs,1953
-documente_shared/domain/entities/processing_case_item.py,sha256=kYw0YmUjZvWv5SOOxqUlS5pf_Z9s3oCnEYFQl2swSAg,10106
+documente_shared/domain/entities/processing_case_item.py,sha256=VbVOqAUHtcCHfAF5VpG47IWkM8t6eSX4VuDifJe7ZzA,10108
 documente_shared/domain/entities/processing_case_item_filters.py,sha256=R_AvDCB496Lww1qn2OwtltqULKE3IpcJB0ejnmRkg7Q,2009
 documente_shared/domain/entities/processing_event.py,sha256=izdBXEz0TMNjxxZVjcM3YclzOv250JOV-amSpqmtQ9c,2180
 documente_shared/domain/entities/scaling.py,sha256=Me1z3X-5NjzPMX-TBQ4xHwE_44tIJegi1QSCHQRNtx4,745
@@ -25,6 +25,7 @@ documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 documente_shared/domain/enums/common.py,sha256=wJWYhh98sdCGL_1WodhYLpoT_IYTzkTDOexclpaIM-0,2827
 documente_shared/domain/enums/document.py,sha256=QwvckW-VJBSujllIVloKlZUh1pI5UnX4oueYbV5CYGw,3205
 documente_shared/domain/enums/processing_case.py,sha256=nIXtya6lmEOB0uWKkH4hLWOeBa8_BgKshHiMrGweswQ,1673
+documente_shared/domain/exceptions.py,sha256=S2xkkFPiR8jVmA6UhRb-bPVrm6JvohXw2MUhZOzdnRo,58
 documente_shared/domain/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/interfaces/scaling.py,sha256=yWAud0rcLKMMf2-QxEE__GRpzqUY0H9_qxlQTfnRYmw,249
 documente_shared/domain/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,12 +43,14 @@ documente_shared/infrastructure/repositories/dynamo_processing_case_item.py,sha2
 documente_shared/infrastructure/repositories/http_document_processing.py,sha256=7n4lHgpN17CCr4_Dz9WRbsXeb4FDMJZNDUshFipA1B8,2179
 documente_shared/infrastructure/repositories/http_processing_case.py,sha256=n1NXJtROTfet2QCKSNnQtLmcNgkJDo6q0Hq48FSpwEU,2249
 documente_shared/infrastructure/repositories/http_processing_case_item.py,sha256=idciRUH3LGbD13TS6BlEpCpC-lWKqf1MssmjcdR3CKU,2971
+documente_shared/infrastructure/repositories/mem_document.py,sha256=jg4rIjgSZijymjY9o7Q1lLcaiW9h-O8j6XljO1bJI7c,1299
+documente_shared/infrastructure/repositories/mem_processing_case.py,sha256=ZAQwp4j0DXQMt92Z-ZR4h9MtbUp9IYy0OHz_sHgkZxY,1147
 documente_shared/infrastructure/s3_bucket.py,sha256=vT_yN42RFQXubtUn8ln-j13Os_-25UGClVtXg5Bkv6I,1932
 documente_shared/infrastructure/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/infrastructure/services/http_scaling.py,sha256=cIo-61nfIwbtO86EGi5r1tFi9g3VJXldhguddt4JUyc,906
 documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
 documente_shared/presentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/presentation/presenters.py,sha256=GGAEwefmjCIVepsUA2oZOVLxXbhhiISPM0Jgt6dT6O0,423
-documente_shared-0.1.96.dist-info/METADATA,sha256=nWhol-EMMx-ilLt_ga76hGdF5bdlDsc4V0_8GcZSahU,920
-documente_shared-0.1.96.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.96.dist-info/RECORD,,
+documente_shared-0.1.98.dist-info/METADATA,sha256=KSDKIiBBkgAI5X4O7h0kad0DhXm9CgLqxCzP3-EVRN0,920
+documente_shared-0.1.98.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.98.dist-info/RECORD,,