documente_shared 0.1.99__py3-none-any.whl → 0.1.100__py3-none-any.whl

documente_shared/domain/entities/in_memory_document.py

@@ -2,10 +2,7 @@ import base64
 from typing import Optional
 from dataclasses import dataclass
 
-from documente_shared.application.files import (
-    remove_slash_from_path,
-    get_filename_from_path,
-)
+from documente_shared.application.files import get_filename_from_path
 from documente_shared.domain.exceptions import InMemoryDocumentContentError
 
 
@@ -15,19 +12,26 @@ class InMemoryDocument(object):
     file_bytes: Optional[bytes] = None
     file_base64: Optional[str] = None
 
+    def __post_init__(self):
+        if not self.file_path:
+            return
+        if self.file_base64 and not self.file_bytes:
+            self.file_bytes = base64.b64decode(self.file_base64)
+        elif self.file_bytes and not self.file_base64:
+            self.file_base64 = base64.b64encode(self.file_bytes).decode()
+
+
     @property
     def is_valid(self) -> bool:
-        return bool(self.file_path) and self.file_bytes
+        return bool(self.file_path and self.file_bytes)
 
     @property
     def has_content(self) -> bool:
-        return bool(self.file_bytes) or bool(self.file_base64)
+        return bool(self.file_bytes or self.file_base64)
 
     @property
     def file_name(self) -> Optional[str]:
-        if not self.file_path:
-            return None
-        return get_filename_from_path(self.file_path)
+        return get_filename_from_path(self.file_path) if self.file_path else None
 
     @property
     def file_key(self) -> Optional[str]:
@@ -46,20 +50,20 @@ class InMemoryDocument(object):
 
     @classmethod
     def from_dict(cls, data: dict):
-        has_bytes_content = data.get('file_bytes') and isinstance(data['file_bytes'], bytes)
-        has_base64_content = data.get('file_base64') and isinstance(data['file_base64'], str)
+        file_bytes = data.get("file_bytes")
+        file_base64 = data.get("file_base64")
 
-        if not has_bytes_content and not has_base64_content:
+        if not file_bytes and not file_base64:
             raise InMemoryDocumentContentError
 
-        if has_bytes_content and not data.get('file_base64'):
-            data['file_base64'] = base64.b64encode(data.get('file_bytes')).decode("utf-8")
+        if file_bytes and not file_base64:
+            file_base64 = base64.b64encode(file_bytes).decode()
 
-        if has_base64_content and not data.get('file_bytes'):
-            data['file_bytes'] = base64.b64decode(data.get('file_base64'))
+        if file_base64 and not file_bytes:
+            file_bytes = base64.b64decode(file_base64)
 
         return cls(
-            file_path=data.get('file_path'),
-            file_bytes=data.get('file_bytes'),
-            file_base64=data.get('file_base64'),
+            file_path=data.get("file_path"),
+            file_bytes=file_bytes,
+            file_base64=file_base64,
         )

{documente_shared-0.1.99.dist-info → documente_shared-0.1.100.dist-info}/METADATA

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

{documente_shared-0.1.99.dist-info → documente_shared-0.1.100.dist-info}/RECORD

@@ -14,7 +14,7 @@ documente_shared/domain/constants.py,sha256=NG5BGaXBr_FnzudjTRPxpDpyiSDdaB_PLCdl
 documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/entities/document.py,sha256=xGZdrqqBkbvDbgoxUvIck6FplXjwLcZzFkcEsuh80u8,12819
 documente_shared/domain/entities/document_metadata.py,sha256=ygyFIC5qwxlm8DUM5kvVFny9zJfPQS8vNLM2br5XsQ8,2353
-documente_shared/domain/entities/in_memory_document.py,sha256=kNszkMfWtmPQHRrARkAqMOUgre-oAPq-bcW-Z-GYoA8,1961
+documente_shared/domain/entities/in_memory_document.py,sha256=5MFG5V4EtNr-wwrLEFZusZPWvla52g5wuI6es5nQ4tM,1999
 documente_shared/domain/entities/processing_case.py,sha256=2wD256_ra5szb8LVmGsauTsFTbkAqT1srOv5zuZQXgA,5640
 documente_shared/domain/entities/processing_case_filters.py,sha256=harKyu7QEuL1bI_Z8_UxkVCMo5r9vHeNHyi_Ja07vjs,1953
 documente_shared/domain/entities/processing_case_item.py,sha256=VbVOqAUHtcCHfAF5VpG47IWkM8t6eSX4VuDifJe7ZzA,10108
@@ -51,6 +51,6 @@ documente_shared/infrastructure/services/http_scaling.py,sha256=cIo-61nfIwbtO86E
 documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
 documente_shared/presentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/presentation/presenters.py,sha256=GGAEwefmjCIVepsUA2oZOVLxXbhhiISPM0Jgt6dT6O0,423
-documente_shared-0.1.99.dist-info/METADATA,sha256=09wpHFc5YW6xO2679vIMvj-UU-VWNjoewbz7YOcgYo4,920
-documente_shared-0.1.99.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.99.dist-info/RECORD,,
+documente_shared-0.1.100.dist-info/METADATA,sha256=1KKQjyPWPCtuChtiVXn5tORQsB1c3wDrH-wgGlTPR_Y,921
+documente_shared-0.1.100.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.100.dist-info/RECORD,,