documente_shared 0.1.115__py3-none-any.whl → 0.1.117__py3-none-any.whl

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/domain/entities/processing_case.py

@@ -27,50 +27,6 @@ class ProcessingCase(object):
     def __post_init__(self):
         self.items = self.items or []
 
-    @property
-    def strategy_id(self) ->str:
-        return str(self.case_type)
-
-    @property
-    def is_procesable(self) -> bool:
-        return self.items and len(self.items) > 0
-
-    @property
-    def is_bcp_microcredito(self) -> bool:
-        return self.case_type and self.case_type.is_bcp_microcredito
-
-    @property
-    def is_univida_soat(self) -> bool:
-        return self.case_type and self.case_type.is_univida_soat
-
-    def pending(self, timezone: tzinfo = la_paz_tz):
-        self.status = ProcessingStatus.PENDING
-        self.started_at = None
-
-    def enqueue(self, timezone: tzinfo = la_paz_tz):
-        self.status = ProcessingStatus.ENQUEUED
-        self.enqueued_at = datetime.now(tz=timezone)
-
-    def processing(self, timezone: tzinfo = la_paz_tz):
-        self.status = ProcessingStatus.PROCESSING
-        self.started_at = datetime.now(tz=timezone)
-
-    def failed(
-        self,
-        error_message: Optional[str] = None,
-        timezone: tzinfo = la_paz_tz,
-    ):
-        self.status = ProcessingStatus.FAILED
-        self.failed_at = datetime.now(tz=timezone)
-
-    def completed(self, timezone: tzinfo = la_paz_tz):
-        self.status = ProcessingStatus.COMPLETED
-        self.completed_at = datetime.now(tz=timezone)
-
-    def deleted(self):
-        self.status = ProcessingStatus.DELETED
-
-    
     def __eq__(self, other: 'ProcessingCase') -> bool:
         if not other:
             return False
@@ -88,6 +44,22 @@ class ProcessingCase(object):
                 and self.metadata == other.metadata
         )
 
+    @property
+    def strategy_id(self) ->str:
+        return str(self.case_type)
+
+    @property
+    def is_procesable(self) -> bool:
+        return self.items and len(self.items) > 0
+
+    @property
+    def is_bcp_microcredito(self) -> bool:
+        return self.case_type and self.case_type.is_bcp_microcredito
+
+    @property
+    def is_univida_soat(self) -> bool:
+        return self.case_type and self.case_type.is_univida_soat
+
     @property
     def to_dict(self) -> dict:
         return {
@@ -130,6 +102,52 @@ class ProcessingCase(object):
     def has_procesable_items(self) -> bool:
         return len(self.procesable_items) > 0
 
+    def pending(self, timezone: tzinfo = la_paz_tz):
+        self.status = ProcessingStatus.PENDING
+        self.started_at = None
+
+    def enqueue(self, timezone: tzinfo = la_paz_tz):
+        self.status = ProcessingStatus.ENQUEUED
+        self.enqueued_at = datetime.now(tz=timezone)
+
+    def processing(self, timezone: tzinfo = la_paz_tz):
+        self.status = ProcessingStatus.PROCESSING
+        self.started_at = datetime.now(tz=timezone)
+
+    def failed(
+        self,
+        error_message: Optional[str] = None,
+        timezone: tzinfo = la_paz_tz,
+    ):
+        self.status = ProcessingStatus.FAILED
+        self.failed_at = datetime.now(tz=timezone)
+
+    def completed(self, timezone: tzinfo = la_paz_tz):
+        self.status = ProcessingStatus.COMPLETED
+        self.completed_at = datetime.now(tz=timezone)
+
+    def deleted(self):
+        self.status = ProcessingStatus.DELETED
+
+    def refresh_status(self):
+        if not self.items:
+            return
+
+        item_statuses = [item.status for item in self.items]
+
+        if any(status == ProcessingStatus.FAILED for status in item_statuses):
+            self.status = ProcessingStatus.INCOMPLETE
+        elif any(status == ProcessingStatus.PROCESSING for status in item_statuses):
+            self.status = ProcessingStatus.PROCESSING
+        elif all(status == ProcessingStatus.COMPLETED for status in item_statuses):
+            self.status = ProcessingStatus.COMPLETED
+        elif all(status == ProcessingStatus.PENDING for status in item_statuses):
+            self.status = ProcessingStatus.PENDING
+        else:
+            self.status = ProcessingStatus.PENDING
+
+
+
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingCase':
         return cls(
@@ -161,3 +179,4 @@ class ProcessingCase(object):
             for item_dict in data.get('items', [])
         ]
         return instance
+

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.115.dist-info → documente_shared-0.1.117.dist-info}/METADATA

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

{documente_shared-0.1.115.dist-info → documente_shared-0.1.117.dist-info}/RECORD

@@ -16,10 +16,10 @@ 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=9X6XCE-zTm0q8ASBn-5TAvNwK4iO_70KHkcgyMSO9JI,13044
 documente_shared/domain/entities/document_metadata.py,sha256=ygyFIC5qwxlm8DUM5kvVFny9zJfPQS8vNLM2br5XsQ8,2353
-documente_shared/domain/entities/in_memory_document.py,sha256=BswTG1LjrQ8nP3pb0T2bSODPjJ2ud4ifbnYxRdFlIhY,2116
-documente_shared/domain/entities/processing_case.py,sha256=NwGGGMyXnNprtO1lCP6rkPbhGhZ6C_z30OeHi_LHXDY,5706
+documente_shared/domain/entities/in_memory_document.py,sha256=8JfVxbgqxkU3QvsuWzjpSp-end6JfffYDUSvMvrQsps,2284
+documente_shared/domain/entities/processing_case.py,sha256=6_1J50jyb8XcOiXvGKDJxyLZ6we-RDOmdimalc3h7oM,6451
 documente_shared/domain/entities/processing_case_filters.py,sha256=harKyu7QEuL1bI_Z8_UxkVCMo5r9vHeNHyi_Ja07vjs,1953
-documente_shared/domain/entities/processing_case_item.py,sha256=KtH5JpJOcGLzVGQrcIGQY7CNWTwkm3_sriHM0ROF7SM,10255
+documente_shared/domain/entities/processing_case_item.py,sha256=Kqozxf91C0SxOUl1ZbEU6_63Lz16mX4RfDqZ5Kf8mm4,10921
 documente_shared/domain/entities/processing_case_item_filters.py,sha256=R_AvDCB496Lww1qn2OwtltqULKE3IpcJB0ejnmRkg7Q,2009
 documente_shared/domain/entities/processing_documents.py,sha256=YYuTkdCNkqlO8cA0onJsZYtstxGt9M5NMuIO_87lB14,352
 documente_shared/domain/entities/processing_event.py,sha256=izdBXEz0TMNjxxZVjcM3YclzOv250JOV-amSpqmtQ9c,2180
@@ -55,6 +55,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.115.dist-info/METADATA,sha256=Jej_jT_-7b__6xiNpA4oiOdYUTPVmn-Z8haN3MIfxOI,963
-documente_shared-0.1.115.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.115.dist-info/RECORD,,
+documente_shared-0.1.117.dist-info/METADATA,sha256=58ttP8rKDPpwFiwi_qYPn3lu64ytZ0HWsz7v-wQyY2g,963
+documente_shared-0.1.117.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.117.dist-info/RECORD,,