documente_shared 0.1.77__tar.gz → 0.1.79__tar.gz

{documente_shared-0.1.77 → documente_shared-0.1.79}/PKG-INFO

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

{documente_shared-0.1.77 → documente_shared-0.1.79}/documente_shared/domain/entities/processing_case_item.py

@@ -16,8 +16,8 @@ class ProcessingCaseItem(object):
     case_id: str
     digest: str
     status: ProcessingStatus
-    document_type: ProcessingDocumentType
     document: InMemoryDocument
+    document_type: Optional[ProcessingDocumentType] = None
     uploaded_from: Optional[ProcessingSource] = None
     processed_csv: Optional[InMemoryDocument] = None
     processed_xlsx: Optional[InMemoryDocument] = None
@@ -107,7 +107,10 @@ class ProcessingCaseItem(object):
             'digest': self.digest,
             'status': str(self.status),
             'document': self.document.to_dict,
-            'document_type': str(self.document_type),
+            'document_type': (
+                str(self.document_type)
+                if self.document_type else None
+            ),
             'uploaded_from': (
                 str(self.uploaded_from)
                 if self.uploaded_from else None
@@ -152,8 +155,8 @@ class ProcessingCaseItem(object):
     ):
         instance_properties = properties or [
             'status',
-            'document_type',
             'document',
+            'document_type',
             'uploaded_from',
             'processed_csv',
             'processed_xlsx',
@@ -182,7 +185,10 @@ class ProcessingCaseItem(object):
             digest=data.get('digest'),
             status=ProcessingStatus.from_value(data.get('status')),
             document=InMemoryDocument.from_dict(data.get('document')),
-            document_type=ProcessingDocumentType.from_value(data.get('document_type')),
+            document_type=(
+                ProcessingDocumentType.from_value(data.get('document_type'))
+                if data.get('document_type') else None
+            ),
             uploaded_from=(
                 ProcessingSource.from_value(data.get('uploaded_from'))
                 if data.get('uploaded_from') else None

{documente_shared-0.1.77 → documente_shared-0.1.79}/documente_shared/domain/repositories/processing_case_item.py

@@ -11,6 +11,10 @@ class ProcessingCaseItemRepository(ABC):
     def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
         raise NotImplementedError
 
+    @abstractmethod
+    def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
+        raise NotImplementedError
+
     @abstractmethod
     def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
         raise NotImplementedError

{documente_shared-0.1.77 → documente_shared-0.1.79}/documente_shared/infrastructure/repositories/dynamo_processing_case_item.py

@@ -3,7 +3,7 @@ from typing import Optional, List
 from boto3.dynamodb.conditions import Key
 
 from documente_shared.domain.entities.processing_case_item import ProcessingCaseItem
-from documente_shared.domain.enums.common import ProcessingStatus
+from documente_shared.domain.entities.processing_case_item_filters import ProcessingCaseItemFilters
 from documente_shared.domain.repositories.processing_case_item import ProcessingCaseItemRepository
 
 from documente_shared.infrastructure.dynamo_table import DynamoDBTable
@@ -19,6 +19,12 @@ class DynamoProcessingCaseItemRepository(
             return ProcessingCaseItem.from_dict(item)
         return None
 
+    def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
+        item = self.get(key={'digest': digest})
+        if item:
+            return ProcessingCaseItem.from_dict(item)
+        return None
+
     def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
         self.put(instance.to_simple_dict)
         return instance
@@ -26,10 +32,14 @@ class DynamoProcessingCaseItemRepository(
     def remove(self, instance: ProcessingCaseItem):
         self.delete(key={'case_id': instance.case_id})
 
-    def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCaseItem]:
+    def filter(
+        self,
+        tenant_slug: str,
+        filters: ProcessingCaseItemFilters,
+    ) -> List[ProcessingCaseItem]:
         items = []
 
-        for status in statuses:
+        for status in filters.statuses:
             response = self._table.query(
                 IndexName='status',
                 KeyConditionExpression=Key('status').eq(status.value),
@@ -40,4 +50,4 @@ class DynamoProcessingCaseItemRepository(
         return [
             ProcessingCaseItem.from_dict(item)
             for item in items
-        ]
+        ]

{documente_shared-0.1.77 → documente_shared-0.1.79}/documente_shared/infrastructure/repositories/http_processing_case_item.py

@@ -1,10 +1,8 @@
 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_item import ProcessingCaseItem
 from documente_shared.domain.entities.processing_case_item_filters import ProcessingCaseItemFilters
-from documente_shared.domain.enums.common import ProcessingStatus
 from documente_shared.domain.repositories.processing_case_item import ProcessingCaseItemRepository
 from documente_shared.infrastructure.documente_client import DocumenteClientMixin
 
@@ -21,6 +19,12 @@ class HttpProcessingCaseItemRepository(
             return ProcessingCaseItem.from_dict(response.json())
         return None
 
+    def find_by_digest(self, digest: str) -> Optional[ProcessingCaseItem]:
+        response = self.session.get(f"{self.api_url}/processing-case-items/{digest}/")
+        if response.status_code == 200:
+            return ProcessingCaseItem.from_dict(response.json())
+        return None
+
     def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
         response = self.session.put(
             url=f"{self.api_url}/processing-case-items/{instance.uuid}/",

{documente_shared-0.1.77 → documente_shared-0.1.79}/pyproject.toml

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