documente_shared 0.1.96__tar.gz → 0.1.97__tar.gz

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

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

documente_shared-0.1.97/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.97/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.97}/pyproject.toml

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