documente_shared 0.1.84__py3-none-any.whl → 0.1.86__py3-none-any.whl

documente_shared/application/dates.py

@@ -0,0 +1,7 @@
+import pytz
+from datetime import datetime
+
+
+
+def utc_now() -> datetime:
+    return datetime.now(tz=pytz.utc)

documente_shared/domain/enums/common.py

@@ -1,3 +1,5 @@
+from typing import List
+
 from documente_shared.domain.base_enum import BaseEnum
 
 
@@ -45,6 +47,21 @@ class ProcessingStatus(BaseEnum):
     CANCELLED = 'CANCELLED'
     IN_REVIEW = 'IN_REVIEW'
 
+    @property
+    def procesable_statuses(self) -> List['ProcessingStatus']:
+        return [
+            ProcessingStatus.PENDING,
+            ProcessingStatus.ENQUEUED,
+            ProcessingStatus.PROCESSING,
+        ]
+
+    @property
+    def final_statuses(self) -> List['ProcessingStatus']:
+        return [
+            ProcessingStatus.COMPLETED,
+            ProcessingStatus.FAILED,
+        ]
+
     @property
     def is_pending(self):
         return self == ProcessingStatus.PENDING
@@ -93,3 +110,10 @@ class ProcessingType(BaseEnum):
     @property
     def is_processing_case(self):
         return self == ProcessingType.PROCESSING_CASE
+
+
+class TaskResultStatus(BaseEnum):
+    PENDING = "PENDING"
+    SUCCESS = "SUCCESS"
+    IN_PROGRESS = "IN_PROGRESS"
+    FAILURE = "FAILURE"

documente_shared/domain/enums/document.py

@@ -1,3 +1,5 @@
+from typing import List
+
 from documente_shared.domain.base_enum import BaseEnum
 
 
@@ -12,6 +14,57 @@ class DocumentProcessingStatus(BaseEnum):
     CANCELLED = 'CANCELLED'
     IN_REVIEW = 'IN_REVIEW'
 
+    @property
+    def procesable_statuses(self) -> List['DocumentProcessingStatus']:
+        return [
+            DocumentProcessingStatus.PENDING,
+            DocumentProcessingStatus.ENQUEUED,
+            DocumentProcessingStatus.PROCESSING,
+        ]
+
+    @property
+    def final_statuses(self) -> List['DocumentProcessingStatus']:
+        return [
+            DocumentProcessingStatus.COMPLETED,
+            DocumentProcessingStatus.FAILED,
+        ]
+
+    @property
+    def is_pending(self):
+        return self == DocumentProcessingStatus.PENDING
+
+    @property
+    def is_enqueued(self):
+        return self == DocumentProcessingStatus.ENQUEUED
+
+    @property
+    def is_processing(self):
+        return self == DocumentProcessingStatus.PROCESSING
+
+    @property
+    def is_completed(self):
+        return self == DocumentProcessingStatus.COMPLETED
+
+    @property
+    def is_incomplete(self):
+        return self == DocumentProcessingStatus.INCOMPLETE
+
+    @property
+    def is_failed(self):
+        return self == DocumentProcessingStatus.FAILED
+
+    @property
+    def is_deleted(self):
+        return self == DocumentProcessingStatus.DELETED
+
+    @property
+    def is_cancelled(self):
+        return self == DocumentProcessingStatus.CANCELLED
+
+    @property
+    def is_in_review(self):
+        return self == DocumentProcessingStatus.IN_REVIEW
+
 
 class DocumentProcessingCategory(BaseEnum):
     CIRCULAR = 'CIRCULAR'

documente_shared/infrastructure/documente_client.py

@@ -18,4 +18,5 @@ class DocumenteClientMixin(object):
         return {
             "X-Api-Key": self.api_key,
             "Content-Type": "application/json"
-        }
+        }
+        

documente_shared/infrastructure/lambdas.py

@@ -0,0 +1,14 @@
+import json
+import boto3
+
+
+def invoke_lambda(function_name: str, payload: dict) -> dict | list | None:
+    client = boto3.client('lambda')
+
+    response = client.invoke(
+        FunctionName=function_name,
+        InvocationType='RequestResponse',
+        Payload=json.dumps(payload),
+    )
+
+    return json.loads(response['Payload'].read().decode('utf-8'))

documente_shared/infrastructure/repositories/http_document_processing.py

@@ -0,0 +1,51 @@
+from dataclasses import dataclass
+from typing import List, Optional
+from requests import Response
+
+from documente_shared.application.payloads import camel_to_snake
+from documente_shared.domain.entities.document import DocumentProcessing
+from documente_shared.domain.enums.document import DocumentProcessingStatus, DocumentProcessingCategory
+from documente_shared.domain.repositories.document import DocumentProcessingRepository
+from documente_shared.infrastructure.documente_client import DocumenteClientMixin
+
+
+@dataclass
+class HttpDocumentProcessingRepository(
+    DocumenteClientMixin,
+    DocumentProcessingRepository,
+):
+    def find(self, digest: str) -> Optional[DocumentProcessing]:
+        response = self.session.get(f"{self.api_url}/documents/{digest}/")
+        if response.status_code == 200:
+            return self._build_processing_case(response)
+        return None
+
+    def persist(self, instance: DocumentProcessing) -> DocumentProcessing:
+        response = self.session.put(
+            url=f"{self.api_url}/documents/{instance.digest}/",
+            json=instance.to_simple_dict,
+        )
+        if response.status_code in [200, 201]:
+            raise Exception(f'Error persisting document processing: {response.text}')
+        return self._build_processing_case(response)
+    
+    def remove(self, instance: DocumentProcessing):
+        self.session.delete(f"{self.api_url}/documents/{instance.digest}/")
+        
+    def filter(self, statuses: List[DocumentProcessingStatus]) -> List[DocumentProcessing]:
+        response = self.session.get(f"{self.api_url}/documents/?statuses={statuses}")
+        if response.status_code == 200:
+            raw_response = response.json()
+            return [
+                DocumentProcessing.from_dict(camel_to_snake(item['documentProcessing']))
+                for item in raw_response.get('data', [])
+            ]
+        return []
+    
+    @classmethod
+    def _build_processing_case(cls, response: Response) -> DocumentProcessing:
+        response_json = response.json()
+        instance_data = response_json.get('data', {})
+        return DocumentProcessing.from_dict(camel_to_snake(instance_data))
+    
+    

documente_shared/presentation/presenters.py

@@ -0,0 +1,16 @@
+from dataclasses import dataclass
+
+from documente_shared.application.dates import utc_now
+from documente_shared.domain.enums.common import TaskResultStatus
+
+
+@dataclass
+class TaskResultPresenter(object):
+    status: TaskResultStatus = TaskResultStatus.SUCCESS
+
+    @property
+    def to_dict(self) -> dict:
+        return {
+            "status": str(self.status),
+            "completed_at": utc_now().isoformat(),
+        }

{documente_shared-0.1.84.dist-info → documente_shared-0.1.86.dist-info}/METADATA

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

{documente_shared-0.1.84.dist-info → documente_shared-0.1.86.dist-info}/RECORD

@@ -1,5 +1,6 @@
 documente_shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/application/dates.py,sha256=uExNddWmX9VEX_u420JGoC7fL-ieJ4956I0aw_WX3sQ,109
 documente_shared/application/digest.py,sha256=Um6E8WfFri2_lly4RFWydJyvSfPZGFcOX-opEOzDCWc,172
 documente_shared/application/exceptions.py,sha256=lQM8m7wmI9OTLGva0gd7s7YT7ldaTk_Ln4t32PpzNf8,654
 documente_shared/application/files.py,sha256=ADiWi6Mk3YQGx3boGsDqdb5wk8qmabkGRy7bhNFa1OY,649
@@ -20,24 +21,28 @@ documente_shared/domain/entities/processing_case_item.py,sha256=U_g99AJdlRRJpJ0N
 documente_shared/domain/entities/processing_case_item_filters.py,sha256=-cAQTSWOepMMcGCBg2X3dd0W_8XHuBTlvOB1d-3sVVM,1971
 documente_shared/domain/entities/processing_event.py,sha256=AMkW4dJjW6ss-uvDeWzVMBIJtax8JNWy-zPo1R-TiWY,1963
 documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/enums/common.py,sha256=vXldMUPhhWo0PfTgYwDSjI8bur_lYcImZYiV7yAO7DQ,2262
-documente_shared/domain/enums/document.py,sha256=NltZA1YVgJ7dVfSQdJFIE0ZUGf9Y-nxNXsVQ6GiPLL4,1827
+documente_shared/domain/enums/common.py,sha256=wJWYhh98sdCGL_1WodhYLpoT_IYTzkTDOexclpaIM-0,2827
+documente_shared/domain/enums/document.py,sha256=QwvckW-VJBSujllIVloKlZUh1pI5UnX4oueYbV5CYGw,3205
 documente_shared/domain/enums/processing_case.py,sha256=LhFhcoWlockxcpplsVdC6M2kpXn7sOdzQySf24wFhx8,1572
 documente_shared/domain/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/repositories/document.py,sha256=vJzr6c92gqBzyhaEdjrvnoneKRrWmJ0AsvocPnhxiLU,767
 documente_shared/domain/repositories/processing_case.py,sha256=9jGpnUnXaDgQE1ZKiet7zDVCc7wVvHUrcPOeacebb3s,796
 documente_shared/domain/repositories/processing_case_item.py,sha256=gPZaQCMYlD6vlnEt6cVIqRmi1K-JWvmDx1f74nd97Cs,974
 documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/infrastructure/documente_client.py,sha256=UjVWs9DKa-yhw5DVbcEs8iJxalHOarusVayi_ob6QhE,494
+documente_shared/infrastructure/documente_client.py,sha256=paO66zNelDyA6D6iqTXXFVQ9ERRZoJCGWR3T3hySsaM,503
 documente_shared/infrastructure/dynamo_table.py,sha256=TMQbcuty7wjDMbuhI8PbT0IGXelgELsNTtqTEQeZ824,2112
+documente_shared/infrastructure/lambdas.py,sha256=sGgkw7Mhvuq2TpbW_RNdf5JvQnuzxWYH6gPOVtQ4DtE,357
 documente_shared/infrastructure/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/infrastructure/repositories/dynamo_document.py,sha256=_Yp4gtA-n-hJ2w2wAM5BMCs2Mf46Q2Kq3eHqlxudkL4,1443
 documente_shared/infrastructure/repositories/dynamo_processing_case.py,sha256=IoIHtlaEe4G5TqIV9IvG45a3HRBVHLfNC9sSgQjabUk,1464
 documente_shared/infrastructure/repositories/dynamo_processing_case_item.py,sha256=4guM8V3YfP7kzYcuVWunGJGmXi0kSSUW8otks39g1vs,1754
+documente_shared/infrastructure/repositories/http_document_processing.py,sha256=7n4lHgpN17CCr4_Dz9WRbsXeb4FDMJZNDUshFipA1B8,2179
 documente_shared/infrastructure/repositories/http_processing_case.py,sha256=FcOnzPUu-iWjj6O5syzqN61u6xogMoQMfbSnALqeK-c,2258
 documente_shared/infrastructure/repositories/http_processing_case_item.py,sha256=babWakDZfHxWgB1OUwZsPTr8nhDwC-CDBn-vYtgUtUM,2494
 documente_shared/infrastructure/s3_bucket.py,sha256=vT_yN42RFQXubtUn8ln-j13Os_-25UGClVtXg5Bkv6I,1932
 documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
-documente_shared-0.1.84.dist-info/METADATA,sha256=jukWF48uMxOI5Kdv6KIj0zYlLTpoQw0iHSIlmiiXfSM,881
-documente_shared-0.1.84.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.84.dist-info/RECORD,,
+documente_shared/presentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/presentation/presenters.py,sha256=GGAEwefmjCIVepsUA2oZOVLxXbhhiISPM0Jgt6dT6O0,423
+documente_shared-0.1.86.dist-info/METADATA,sha256=iUB2ODiR3m6PjpgmIqt3l2O5yFcyhGelTcNYOQrjqcY,881
+documente_shared-0.1.86.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.86.dist-info/RECORD,,