documente_shared 0.1.85__py3-none-any.whl → 0.1.87__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/entities/processing_case_filters.py

@@ -17,6 +17,7 @@ class ProcessingCaseFilters(object):
     statuses: List[ProcessingStatus] = None
     case_types: List[ProcessingCaseType] = None
     include_archived: bool = False
+    tenant_slug: Optional[str] = None
 
     def __post_init__(self):
         self.case_ids = self.case_ids or []

documente_shared/domain/entities/processing_case_item_filters.py

@@ -20,6 +20,7 @@ class ProcessingCaseItemFilters(object):
     statuses: List[ProcessingStatus] = None
     document_types: List[ProcessingDocumentType] = None
     include_archived: bool = False
+    tenant_slug: Optional[str] = None
 
     def __post_init__(self):
         self.statuses = self.statuses or []

documente_shared/domain/entities/scaling.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class ScalingRequirements(object):
+    documents: Optional[int] = 0
+    processing_cases: Optional[int] = 0
+    processing_case_items: Optional[int] = 0
+
+    @property
+    def to_dict(self):
+        return {
+            "documents": self.documents,
+            "processing_cases": self.processing_cases,
+            "processing_case_items": self.processing_case_items,
+        }
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "ScalingRequirements":
+        return cls(
+            documents=data.get("documents", 0),
+            processing_cases=data.get("processing_cases", 0),
+            processing_case_items=data.get("processing_case_items", 0),
+        )

documente_shared/domain/enums/common.py

@@ -110,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/interfaces/scaling.py

@@ -0,0 +1,10 @@
+from abc import ABC, abstractmethod
+
+from documente_shared.domain.entities.scaling import ScalingRequirements
+
+
+class ScalingService(ABC):
+
+    @abstractmethod
+    def get_scaling_requirements(self) -> ScalingRequirements:
+        raise NotImplementedError

documente_shared/domain/repositories/processing_case.py

@@ -20,5 +20,5 @@ class ProcessingCaseRepository(ABC):
         raise NotImplementedError
 
     @abstractmethod
-    def filter(self, tenant_slug: str, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
+    def filter(self, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
         raise NotImplementedError

documente_shared/domain/repositories/processing_case_item.py

@@ -26,7 +26,6 @@ class ProcessingCaseItemRepository(ABC):
     @abstractmethod
     def filter(
         self,
-        tenant_slug: str,
         filters: ProcessingCaseItemFilters,
     ) -> List[ProcessingCaseItem]:
         raise NotImplementedError

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/dynamo_processing_case.py

@@ -26,7 +26,7 @@ class DynamoProcessingCaseRepository(
     def remove(self, instance: ProcessingCase):
         self.delete(key={'uuid': instance.uuid})
 
-    def filter(self, tenant_slug: str, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
+    def filter(self, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
         items = []
 
         for status in filters.statuses:

documente_shared/infrastructure/repositories/dynamo_processing_case_item.py

@@ -34,7 +34,6 @@ class DynamoProcessingCaseItemRepository(
 
     def filter(
         self,
-        tenant_slug: str,
         filters: ProcessingCaseItemFilters,
     ) -> List[ProcessingCaseItem]:
         items = []
@@ -50,4 +49,4 @@ class DynamoProcessingCaseItemRepository(
         return [
             ProcessingCaseItem.from_dict(item)
             for item in items
-        ]
+        ]

documente_shared/infrastructure/repositories/http_processing_case.py

@@ -34,11 +34,11 @@ class HttpProcessingCaseRepository(
     def remove(self, instance: ProcessingCase):
         self.session.delete(f"{self.api_url}/v1/processing-cases/{instance.uuid}/")
 
-    def filter(self, tenant_slug: str, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
+    def filter(self, filters: ProcessingCaseFilters) -> List[ProcessingCase]:
         response = self.session.get(
             url=f"{self.api_url}/v1/processing-cases/",
             headers={
-                "X-Tenant": tenant_slug,
+                "X-Tenant": filters.tenant_slug,
             }
         )
         if response.status_code == 200:
@@ -49,6 +49,7 @@ class HttpProcessingCaseRepository(
             ]
         return []
 
+
     @classmethod
     def _build_processing_case(cls, response: Response) -> ProcessingCase:
         response_json = response.json()

documente_shared/infrastructure/repositories/http_processing_case_item.py

@@ -40,6 +40,18 @@ class HttpProcessingCaseItemRepository(
         self.session.delete(f"{self.api_url}/v1/processing-case-items/{instance.uuid}/")
 
     def filter(
+        self,
+        filters: ProcessingCaseItemFilters,
+    ) -> List[ProcessingCaseItem]:
+        response = self.session.get(f"{self.api_url}/v1/processing-case-items/")
+        if response.status_code == 200:
+            raw_response = response.json()
+            return [
+                ProcessingCaseItem.from_dict(camel_to_snake(item_data))
+                for item_data in raw_response.get('data', [])
+            ]
+        return []
+    def filter_with_tenant(
         self,
         tenant_slug: str,
         filters: ProcessingCaseItemFilters,

documente_shared/infrastructure/services/http_scaling.py

@@ -0,0 +1,24 @@
+from dataclasses import dataclass
+
+from loguru import logger
+
+from documente_shared.application.payloads import camel_to_snake
+from documente_shared.domain.entities.scaling import ScalingRequirements
+from documente_shared.domain.interfaces.scaling import ScalingService
+from documente_shared.infrastructure.documente_client import DocumenteClientMixin
+
+
+@dataclass
+class HttpScalingService(
+    DocumenteClientMixin,
+    ScalingService,
+):
+    def get_scaling_requirements(self) -> ScalingRequirements:
+        response = self.session.get(f"{self.api_url}/v1/scaling-requirements/")
+        if response.status_code == 200:
+            return ScalingRequirements.from_dict(
+                data=camel_to_snake(response.json())
+            )
+
+        logger.warning(f'Error getting scaling requirements: {response.text}')
+        return ScalingRequirements()

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.85
+Version: 0.1.87
 Summary: Shared utilities for Documente AI projects
 License: MIT
 Author: Tech
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: boto3 (>=1.37.19,<2.0.0)
 Requires-Dist: botocore (>=1.37.19,<2.0.0)
+Requires-Dist: loguru (>=0.7.3,<0.8.0)
 Requires-Dist: pytz (>=2025.2,<2026.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: sentry-sdk (>=2.19.2,<3.0.0)

{documente_shared-0.1.85.dist-info → documente_shared-0.1.87.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
@@ -15,30 +16,38 @@ documente_shared/domain/entities/document.py,sha256=AthTUyA-QZE3WAT7lMoKVr_Z8mO_
 documente_shared/domain/entities/document_metadata.py,sha256=ygyFIC5qwxlm8DUM5kvVFny9zJfPQS8vNLM2br5XsQ8,2353
 documente_shared/domain/entities/in_memory_result.py,sha256=0sLNUrovKFQx4M-E9e4DrAiVgch2i4AKA-9BQBRaeI8,1482
 documente_shared/domain/entities/processing_case.py,sha256=_UTMCSQjTSttNWa-NC7eWJITkNm6NZuGGT9gNm9D1mA,4970
-documente_shared/domain/entities/processing_case_filters.py,sha256=FgyxB4mQb0nEGjIbUB9OiazkKL4yHRRC6bvmjD5NT8k,1915
+documente_shared/domain/entities/processing_case_filters.py,sha256=harKyu7QEuL1bI_Z8_UxkVCMo5r9vHeNHyi_Ja07vjs,1953
 documente_shared/domain/entities/processing_case_item.py,sha256=U_g99AJdlRRJpJ0NPfmXzlr2x04l5hc53oNLxx_Q7XQ,9734
-documente_shared/domain/entities/processing_case_item_filters.py,sha256=-cAQTSWOepMMcGCBg2X3dd0W_8XHuBTlvOB1d-3sVVM,1971
+documente_shared/domain/entities/processing_case_item_filters.py,sha256=R_AvDCB496Lww1qn2OwtltqULKE3IpcJB0ejnmRkg7Q,2009
 documente_shared/domain/entities/processing_event.py,sha256=AMkW4dJjW6ss-uvDeWzVMBIJtax8JNWy-zPo1R-TiWY,1963
+documente_shared/domain/entities/scaling.py,sha256=Me1z3X-5NjzPMX-TBQ4xHwE_44tIJegi1QSCHQRNtx4,745
 documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/enums/common.py,sha256=yoq__M53bCCf7B2tbOGdOpieYvGpiJ69s6T0FqvLUII,2687
+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/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/domain/interfaces/scaling.py,sha256=0Mefc_GVNdyRtAdkxlnvQuXrmM4Yx7VOJQCfVZOgISQ,257
 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/domain/repositories/processing_case.py,sha256=QcY0LumRokRLmL3IWkOZTgN-LQ-Kku5_v7DWujO1Dfw,778
+documente_shared/domain/repositories/processing_case_item.py,sha256=uBgJN2fJnZDcKUk018P_Fv25dQRZD5FBxFxsCQNaGAQ,948
 documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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/dynamo_processing_case.py,sha256=jkTVHThKHshLI53OV7ivK-WchFoAZTnaXlgh_1OX52k,1446
+documente_shared/infrastructure/repositories/dynamo_processing_case_item.py,sha256=B2ElsASpXNRkwwjdCqXyvDU-LBrLNdwPfHLMvvG9c-Y,1729
 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/repositories/http_processing_case.py,sha256=n1NXJtROTfet2QCKSNnQtLmcNgkJDo6q0Hq48FSpwEU,2249
+documente_shared/infrastructure/repositories/http_processing_case_item.py,sha256=cCi7uo1aHPhroxRRlFOhIUgGkY_S2dvWl1DAVr_maxw,2966
 documente_shared/infrastructure/s3_bucket.py,sha256=vT_yN42RFQXubtUn8ln-j13Os_-25UGClVtXg5Bkv6I,1932
+documente_shared/infrastructure/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/infrastructure/services/http_scaling.py,sha256=SK0vReNGXzfN2yKNiYq9u1fvDECU5dHynq5507i7fgI,856
 documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
-documente_shared-0.1.85.dist-info/METADATA,sha256=On4dTXDR9MF1K2P2FjhW_OeCNOLfBQW0XZT2hI8t5Zg,881
-documente_shared-0.1.85.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.85.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.87.dist-info/METADATA,sha256=zTsM418cXINfRJx1HAF_JUXqd7Pr8mYEkcuXPt8Ah5A,920
+documente_shared-0.1.87.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.87.dist-info/RECORD,,