documente_shared 0.1.20__tar.gz → 0.1.22__tar.gz

{documente_shared-0.1.20 → documente_shared-0.1.22}/PKG-INFO

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

documente_shared-0.1.22/documente_shared/application/time_utils.py

@@ -0,0 +1,10 @@
+from datetime import datetime
+from typing import Union
+
+
+def get_datetime_from_data(input_datetime: Union[datetime, str]):
+    if isinstance(input_datetime, datetime):
+        return input_datetime
+    elif isinstance(input_datetime, str):
+        return datetime.fromisoformat(input_datetime)
+    return None

{documente_shared-0.1.20 → documente_shared-0.1.22}/documente_shared/domain/entities.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from decimal import Decimal
 from typing import Optional, List
 
+from documente_shared.application.time_utils import get_datetime_from_data
 from documente_shared.domain.enums import (
     DocumentProcessStatus,
     DocumentProcessSubCategory,
@@ -76,9 +77,13 @@ class DocumentProcess(object):
     def deleted(self):
         self.status = DocumentProcessStatus.DELETED
 
+    @property
+    def extended_filename(self) -> str:
+        return self.file_path.split('/')[-1]
+
     @property
     def filename(self) -> str:
-        filename_with_extension = self.file_path.split('/')[-1]
+        filename_with_extension = self.extended_filename
         return filename_with_extension.split('.')[0]
 
     @property
@@ -146,24 +151,9 @@ class DocumentProcess(object):
                 Decimal(data.get('processing_time'))
                 if data.get('processing_time') else None
             ),
-            uploaded_at=(
-                datetime.fromisoformat(data.get('uploaded_at'))
-                if data.get('uploaded_at') else None
-            ),
-            enqueued_at=(
-                datetime.fromisoformat(data.get('enqueued_at'))
-                if data.get('enqueued_at') else None
-            ),
-            started_at=(
-                datetime.fromisoformat(data.get('started_at'))
-                if data.get('started_at') else None
-            ),
-            failed_at=(
-                datetime.fromisoformat(data.get('failed_at'))
-                if data.get('failed_at') else None
-            ),
-            completed_at=(
-                datetime.fromisoformat(data.get('completed_at'))
-                if data.get('processed_at') else None
-            ),
+            uploaded_at=get_datetime_from_data(input_datetime=data.get('uploaded_at')),
+            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')),
+            completed_at=get_datetime_from_data(input_datetime=data.get('completed_at')),
         )

{documente_shared-0.1.20 → documente_shared-0.1.22}/pyproject.toml

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