documente_shared 0.1.116__tar.gz → 0.1.118__tar.gz

{documente_shared-0.1.116 → documente_shared-0.1.118}/PKG-INFO

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

{documente_shared-0.1.116 → documente_shared-0.1.118}/documente_shared/domain/entities/document.py

@@ -278,6 +278,11 @@ class DocumentProcessing(object):
         simple_dict.pop('metadata_items')
         return simple_dict
 
+    @property
+    def to_queue_dict(self) -> dict:
+        return self.to_dict
+
+
     def overload(
         self,
         new_instance: 'DocumentProcessing',

{documente_shared-0.1.116 → documente_shared-0.1.118}/documente_shared/domain/entities/in_memory_document.py

@@ -2,7 +2,7 @@ import base64
 from typing import Optional
 from dataclasses import dataclass
 
-from documente_shared.application.files import get_filename_from_path, remove_extension
+from documente_shared.application.files import get_filename_from_path, remove_extension, remove_slash_from_path
 from documente_shared.domain.exceptions import InMemoryDocumentContentError
 
 
@@ -39,7 +39,7 @@ class InMemoryDocument(object):
 
     @property
     def file_key(self) -> Optional[str]:
-        return self.file_name
+        return remove_slash_from_path(self.file_path)
 
     @property
     def is_procesable(self) -> bool:
@@ -52,6 +52,12 @@ class InMemoryDocument(object):
             "file_base64": self.file_base64,
         }
 
+    @property
+    def to_queue_dict(self) -> dict:
+        return {
+            "file_path": self.file_path,
+        }
+
     @classmethod
     def from_dict(cls, data: dict):
         file_bytes = data.get("file_bytes")

{documente_shared-0.1.116 → documente_shared-0.1.118}/documente_shared/domain/entities/processing_case_item.py

@@ -210,6 +210,27 @@ class ProcessingCaseItem(object):
         simple_dict = self.to_dict.copy()
         return simple_dict
 
+    @property
+    def to_queue_dict(self) -> dict:
+        queue_dict = self.to_dict.copy()
+        queue_dict["document"] = (
+            self.document.to_queue_dict
+            if self.document else None
+        )
+        queue_dict["processed_csv"] = (
+            self.processed_csv.to_queue_dict
+            if self.processed_csv else None
+        )
+        queue_dict["processed_xlsx"] = (
+            self.processed_xlsx.to_queue_dict
+            if self.processed_xlsx else None
+        )
+        queue_dict["processed_json"] = (
+            self.processed_json.to_queue_dict
+            if self.processed_json else None
+        )
+        return queue_dict
+
     @property
     def to_persist_dict(self) -> dict:
         simple_dict = self.to_dict.copy()

{documente_shared-0.1.116 → documente_shared-0.1.118}/documente_shared/domain/entities/processing_event.py

@@ -47,6 +47,12 @@ class ProcessingEvent(object):
             'timestamp': self.timestamp.isoformat() if self.timestamp else None,
         }
 
+    @property
+    def to_queue_dict(self) -> dict:
+        dict_data = self.to_dict
+        dict_data['instance'] = self.instance.to_queue_dict
+        return dict_data
+
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingEvent':
         processing_type = ProcessingType.from_value(data.get('processing_type'))

{documente_shared-0.1.116 → documente_shared-0.1.118}/pyproject.toml

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