documente_shared 0.1.50__tar.gz → 0.1.51__tar.gz

{documente_shared-0.1.50 → documente_shared-0.1.51}/PKG-INFO

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

documente_shared-0.1.51/documente_shared/domain/constants.py

@@ -0,0 +1,3 @@
+import pytz
+
+la_paz_tz = pytz.timezone("America/La_Paz")

{documente_shared-0.1.50 → documente_shared-0.1.51}/documente_shared/domain/entities/document.py

@@ -1,9 +1,10 @@
 from dataclasses import dataclass
-from datetime import datetime
+from datetime import datetime, tzinfo
 from decimal import Decimal
 from typing import Optional, List
 
 from documente_shared.application.time_utils import get_datetime_from_data
+from documente_shared.domain.constants import la_paz_tz
 from documente_shared.domain.entities.document_metadata import DocumentProcessingMetadata
 from documente_shared.domain.enums import (
     DocumentProcessingStatus,
@@ -79,22 +80,26 @@ class DocumentProcessing(object):
             DocumentProcessingStatus.FAILED,
         ]
 
-    def enqueue(self):
+    def enqueue(self, timezone: tzinfo = la_paz_tz):
         self.status = DocumentProcessingStatus.ENQUEUED
-        self.enqueued_at = datetime.now()
+        self.enqueued_at = datetime.now(tz=timezone)
 
-    def processing(self):
+    def processing(self, timezone: tzinfo = la_paz_tz):
         self.status = DocumentProcessingStatus.PROCESSING
-        self.started_at = datetime.now()
+        self.started_at = datetime.now(tz=timezone)
 
-    def failed(self, error_message: Optional[str] = None):
+    def failed(
+        self,
+        error_message: Optional[str] = None,
+        timezone: tzinfo = la_paz_tz,
+    ):
         self.failed_reason = error_message
         self.status = DocumentProcessingStatus.FAILED
-        self.failed_at = datetime.now()
+        self.failed_at = datetime.now(tz=timezone)
 
-    def completed(self):
+    def completed(self, timezone: tzinfo = la_paz_tz):
         self.status = DocumentProcessingStatus.COMPLETED
-        self.completed_at = datetime.now()
+        self.completed_at = datetime.now(tz=timezone)
 
     def deleted(self):
         self.status = DocumentProcessingStatus.DELETED

{documente_shared-0.1.50 → documente_shared-0.1.51}/pyproject.toml

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