documente_shared 0.1.35__py3-none-any.whl → 0.1.37__py3-none-any.whl

documente_shared/domain/entities/document_process.py

@@ -77,7 +77,8 @@ class DocumentProcess(object):
         self.status = DocumentProcessStatus.PROCESSING
         self.started_at = datetime.now()
 
-    def failed(self):
+    def failed(self, error_message: Optional[str] = None):
+        self.failed_reason = error_message
         self.status = DocumentProcessStatus.FAILED
         self.failed_at = datetime.now()
 

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

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

{documente_shared-0.1.35.dist-info → documente_shared-0.1.37.dist-info}/RECORD

@@ -5,15 +5,15 @@ documente_shared/application/time_utils.py,sha256=XDH27cKgoTFO8ad1JgrxKaeT7sZ1fd
 documente_shared/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/base_enum.py,sha256=DojAfn-zQdtjtImeHUpBzE6TBTm07XrbMOdW3h8RVd8,1449
 documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/entities/document_process.py,sha256=WtDz1WMVVgn2FvmTDRnQNVLl90Sp06P-ta4p8RTddfQ,7436
+documente_shared/domain/entities/document_process.py,sha256=vDcri1NahlrTtE7wDStmMa135fqQZj2msW9xIQ0rF9E,7516
 documente_shared/domain/entities/document_process_metadata.py,sha256=rBHkDkoKwrxReKtXIScU1vrCO_6bg2LwWhrtXMcQ8TA,2351
 documente_shared/domain/enums.py,sha256=s3bFDkpplWvJWNpUwPOkC9a3OYUYsx8uTusN_FrzNQk,463
-documente_shared/domain/repositories.py,sha256=KEYbJKsfrRkJBCIjXQA3S5jdNqE5lUcQuvHiZig4B6M,526
+documente_shared/domain/repositories.py,sha256=o9_ty3NPxCjb4hHIZbk71ECLqTjoKQI5MKWQje1KEyk,738
 documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/infrastructure/dynamo_repositories.py,sha256=WVyubQO8pUwm_y6VYU2n3fN4eofBXLBRzY1oehotp80,784
-documente_shared/infrastructure/dynamo_table.py,sha256=Lod8vwUCGX65SNH_8hvlpG6rRUYb4jfZqoFHD1YloAA,2096
+documente_shared/infrastructure/dynamo_repositories.py,sha256=w_AW3IACVxRU7A3mNma2Elpbx7naRV4wY5mChW1WmZQ,1394
+documente_shared/infrastructure/dynamo_table.py,sha256=dK05KgFvIYCmOdMpq9-OV_OBrP6cCngiUikCJrxlwt4,2112
 documente_shared/infrastructure/s3_bucket.py,sha256=Nf4bHSC3TJeaKvOQSVtV1hG8dWqEZZnA-JWU0PNTw24,1940
 documente_shared/infrastructure/sqs_queue.py,sha256=PSiTAnjXvQ-W-9mzLpH2UjbQJTvYkMiaxNaMecF-cR4,1505
-documente_shared-0.1.35.dist-info/METADATA,sha256=G_XmvO8Sciyg2UqOFU1YsXw-kPnDacy8Ih2fspFIOdI,640
-documente_shared-0.1.35.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.35.dist-info/RECORD,,
+documente_shared-0.1.37.dist-info/METADATA,sha256=7nRGwFRypTAO5ZW-hVLHkaJy6M0vm6c1-n5YpbiK8fs,640
+documente_shared-0.1.37.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.37.dist-info/RECORD,,