documente_shared 0.1.36__tar.gz → 0.1.38__tar.gz

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

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

{documente_shared-0.1.36 → documente_shared-0.1.38}/documente_shared/domain/entities/document_process.py

@@ -69,6 +69,13 @@ class DocumentProcess(object):
             self.file_path,
         ])
 
+    @property
+    def is_finished(self) -> bool:
+        return self.status in [
+            DocumentProcessStatus.COMPLETED,
+            DocumentProcessStatus.FAILED,
+        ]
+
     def enqueue(self):
         self.status = DocumentProcessStatus.ENQUEUED
         self.enqueued_at = datetime.now()
@@ -114,6 +121,22 @@ class DocumentProcess(object):
         filename_with_extension = self.extended_filename
         return filename_with_extension.split('.')[0]
 
+    def __eq__(self, other: 'DocumentProcess') -> bool:
+        if not other:
+            return False
+
+        return (
+            self.digest == other.digest
+            and self.status == other.status
+            and self.file_path == other.file_path
+            and self.uploaded_at == other.uploaded_at
+            and self.enqueued_at == other.enqueued_at
+            and self.started_at == other.started_at
+            and self.failed_at == other.failed_at
+            and self.completed_at == other.completed_at
+        )
+
+
     @property
     def to_dict(self) -> dict:
         return {

{documente_shared-0.1.36 → documente_shared-0.1.38}/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.38}/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.38}/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.38}/pyproject.toml

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