documente_shared 0.1.36__tar.gz → 0.1.37__tar.gz

{documente_shared-0.1.36 → documente_shared-0.1.37}/PKG-INFO

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

{documente_shared-0.1.36 → documente_shared-0.1.37}/documente_shared/domain/repositories.py

@@ -1,7 +1,8 @@
 from abc import ABC, abstractmethod
-from typing import Optional
+from typing import Optional, List
 
 from documente_shared.domain.entities.document_process import DocumentProcess
+from documente_shared.domain.enums import DocumentProcessStatus
 
 
 class DocumentProcessRepository(ABC):
@@ -16,4 +17,9 @@ class DocumentProcessRepository(ABC):
 
     @abstractmethod
     def remove(self, instance: DocumentProcess):
+        raise NotImplementedError
+
+
+    @abstractmethod
+    def filter(self, statuses: List[DocumentProcessStatus]) -> List[DocumentProcess]:
         raise NotImplementedError

{documente_shared-0.1.36 → documente_shared-0.1.37}/documente_shared/infrastructure/dynamo_repositories.py

@@ -1,6 +1,9 @@
-from typing import Optional
+from typing import Optional, List
+
+from boto3.dynamodb.conditions import Key
 
 from documente_shared.domain.entities.document_process import DocumentProcess
+from documente_shared.domain.enums import DocumentProcessStatus
 from documente_shared.domain.repositories import DocumentProcessRepository
 from documente_shared.infrastructure.dynamo_table import DynamoDBTable
 
@@ -21,4 +24,20 @@ class DynamoDocumentProcessRepository(
         return instance
 
     def remove(self, instance: DocumentProcess):
-        self.delete(key={'digest': instance.digest})
+        self.delete(key={'digest': instance.digest})
+
+    def filter(self, statuses: List[DocumentProcessStatus]) -> List[DocumentProcess]:
+        items = []
+
+        for status in statuses:
+            response = self._table.query(
+                IndexName='status',
+                KeyConditionExpression=Key('status').eq(status.value),
+            )
+            status_items = response.get('Items', [])
+            items.extend(status_items)
+
+        return [
+            DocumentProcess.from_dict(item)
+            for item in items
+        ]

{documente_shared-0.1.36 → documente_shared-0.1.37}/documente_shared/infrastructure/dynamo_table.py

@@ -23,8 +23,8 @@ class DynamoDBTable(object):
         return self.put({**key, **attributes})
 
 
-    def filter_by(self, attribute, target_value):
-        return self._table(
+    def filter_by(self, attribute: str, target_value: str):
+        return self._table.query(
             FilterExpression=Key(attribute).eq(target_value),
         ).get('Items')
 

{documente_shared-0.1.36 → documente_shared-0.1.37}/pyproject.toml

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