documente_shared 0.1.33__tar.gz → 0.1.37__tar.gz

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

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

{documente_shared-0.1.33 → documente_shared-0.1.37}/documente_shared/domain/entities/document_process.py

@@ -34,6 +34,7 @@ class DocumentProcess(object):
     enqueued_at: Optional[datetime] = None
     started_at: Optional[datetime] = None
     failed_at: Optional[datetime] = None
+    failed_reason: Optional[str] = None
     completed_at: Optional[datetime] = None
     metadata_items: Optional[List[DocumentProcessMetadata]] = None
 
@@ -76,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()
 
@@ -137,6 +139,7 @@ class DocumentProcess(object):
             'enqueued_at': self.enqueued_at.isoformat() if self.enqueued_at else None,
             'started_at': self.started_at.isoformat() if self.started_at else None,
             'failed_at': self.failed_at.isoformat() if self.failed_at else None,
+            'failed_reason': self.failed_reason,
             'completed_at': self.completed_at.isoformat() if self.completed_at else None,
             'metadata_items': [metadata.to_dict for metadata in self.metadata_items],
         }
@@ -156,7 +159,7 @@ class DocumentProcess(object):
             'status', 'metadata', 'file_path', 'file_bytes', 'category', 'sub_category',
             'processed_csv_path', 'processed_csv_bytes', 'processed_xlsx_path', 'processed_metadata_path',
             'processed_xlsx_bytes', 'processing_time', 'uploaded_at',
-            'enqueued_at', 'started_at', 'failed_at', 'completed_at',
+            'enqueued_at', 'started_at', 'failed_at', 'failed_reason', 'completed_at',
         ]
         for _property in instance_properties:
             property_value = getattr(new_instance, _property)
@@ -190,6 +193,7 @@ class DocumentProcess(object):
             enqueued_at=get_datetime_from_data(input_datetime=data.get('enqueued_at')),
             started_at=get_datetime_from_data(input_datetime=data.get('started_at')),
             failed_at=get_datetime_from_data(input_datetime=data.get('failed_at')),
+            failed_reason=data.get('failed_reason'),
             completed_at=get_datetime_from_data(input_datetime=data.get('completed_at')),
             metadata_items=[
                 DocumentProcessMetadata.from_dict(metadata)

{documente_shared-0.1.33 → 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.33 → 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.33 → 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.33 → documente_shared-0.1.37}/pyproject.toml

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