documente_shared 0.1.85__tar.gz → 0.1.87__tar.gz

{documente_shared-0.1.85 → documente_shared-0.1.87}/PKG-INFO

@@ -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.87/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-0.1.85 → documente_shared-0.1.87}/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-0.1.85 → documente_shared-0.1.87}/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-0.1.87/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-0.1.85 → documente_shared-0.1.87}/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-0.1.87/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-0.1.85 → documente_shared-0.1.87}/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-0.1.85 → documente_shared-0.1.87}/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-0.1.87/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-0.1.85 → documente_shared-0.1.87}/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-0.1.85 → documente_shared-0.1.87}/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-0.1.85 → documente_shared-0.1.87}/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-0.1.85 → documente_shared-0.1.87}/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-0.1.87/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-0.1.87/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 → documente_shared-0.1.87}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.85"
+version = "0.1.87"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"
@@ -13,6 +13,7 @@ botocore = "^1.37.19"
 sentry-sdk = "^2.19.2"
 pytz = "^2025.2"
 requests = "^2.32.3"
+loguru = "^0.7.3"
 
 [tool.poetry.dev-dependencies]
 pytest = "^8.3.4"