documente_shared 0.1.72__tar.gz → 0.1.73__tar.gz

{documente_shared-0.1.72 → documente_shared-0.1.73}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.72
+Version: 0.1.73
 Summary: Shared utilities for Documente AI projects
 License: MIT
 Author: Tech
@@ -14,6 +14,7 @@ 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: pytz (>=2025.2,<2026.0)
+Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: sentry-sdk (>=2.19.2,<3.0.0)
 Description-Content-Type: text/markdown
 

documente_shared-0.1.73/documente_shared/domain/constants.py

@@ -0,0 +1,8 @@
+import pytz
+from os import environ
+
+
+la_paz_tz = pytz.timezone("America/La_Paz")
+
+DOCUMENTE_API_URL = environ.get("DOCUMENTE_API_URL")
+DOCUMENTE_API_KEY = environ.get("DOCUMENTE_API_KEY")

{documente_shared-0.1.72 → documente_shared-0.1.73}/documente_shared/domain/entities/processing_case.py

@@ -6,15 +6,16 @@ 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.processing_case_item import ProcessingCaseItem
 from documente_shared.domain.enums.common import ProcessingStatus
-from documente_shared.domain.enums.processing_case import ProcessingCaseCategory
+from documente_shared.domain.enums.processing_case import ProcessingCaseType
 
 
 @dataclass
 class ProcessingCase(object):
     uuid: str
-    label: str
+    name: str
+    tenant_slug: str
     status: ProcessingStatus
-    category: Optional[ProcessingCaseCategory] = None
+    case_type: Optional[ProcessingCaseType] = None
     enqueued_at: Optional[datetime] = None
     started_at: Optional[datetime] = None
     failed_at: Optional[datetime] = None
@@ -63,9 +64,9 @@ class ProcessingCase(object):
 
         return (
                 self.uuid == other.uuid
-                and self.label == other.label
+                and self.name == other.name
                 and self.status == other.status
-                and self.category == other.category
+                and self.case_type == other.case_type
                 and self.enqueued_at == other.enqueued_at
                 and self.started_at == other.started_at
                 and self.failed_at == other.failed_at
@@ -78,11 +79,11 @@ class ProcessingCase(object):
     def to_dict(self) -> dict:
         return {
             'uuid': self.uuid,
-            'label': self.label,
+            'label': self.name,
             'status': str(self.status),
             'category': (
-                str(self.category)
-                if self.category else None
+                str(self.case_type)
+                if self.case_type else None
             ),
             'enqueued_at': self.enqueued_at.isoformat() if self.enqueued_at else None,
             'started_at': self.started_at.isoformat() if self.started_at else None,
@@ -125,10 +126,10 @@ class ProcessingCase(object):
     def from_dict(cls, data: dict) -> 'ProcessingCase':
         return cls(
             uuid=data.get('uuid'),
-            label=data.get('label'),
+            name=data.get('label'),
             status=ProcessingStatus.from_value(data.get('status')),
-            category=(
-                ProcessingCaseCategory.from_value(data.get('category'))
+            case_type=(
+                ProcessingCaseType.from_value(data.get('category'))
                 if data.get('category') else None
             ),
             enqueued_at=get_datetime_from_data(input_datetime=data.get('enqueued_at')),

{documente_shared-0.1.72 → documente_shared-0.1.73}/documente_shared/domain/enums/processing_case.py

@@ -1,12 +1,13 @@
 from documente_shared.domain.base_enum import BaseEnum
 
 
-class ProcessingCaseCategory(BaseEnum):
+class ProcessingCaseType(BaseEnum):
     BCP_MICROCREDITO = 'BCP_MICROCREDITO'
+    UNIVIDA_SOAT = 'UNIVIDA_SOAT'
 
     @property
     def is_bcp_microcredito(self):
-        return self == ProcessingCaseCategory.BCP_MICROCREDITO
+        return self == ProcessingCaseType.BCP_MICROCREDITO
 
 
 class ProcessingDocumentType(BaseEnum):

{documente_shared-0.1.72 → documente_shared-0.1.73}/documente_shared/domain/repositories/processing_case.py

@@ -8,7 +8,7 @@ from documente_shared.domain.enums.common import ProcessingStatus
 class ProcessingCaseRepository(ABC):
 
     @abstractmethod
-    def find(self, case_id: str) -> Optional[ProcessingCase]:
+    def find(self, uuid: str) -> Optional[ProcessingCase]:
         raise NotImplementedError
 
     @abstractmethod

{documente_shared-0.1.72 → documente_shared-0.1.73}/documente_shared/domain/repositories/processing_case_item.py

@@ -9,7 +9,7 @@ from documente_shared.domain.enums.common import ProcessingStatus
 class ProcessingCaseItemRepository(ABC):
 
     @abstractmethod
-    def find(self, digest: str) -> Optional[ProcessingCaseItem]:
+    def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
         raise NotImplementedError
 
     @abstractmethod

documente_shared-0.1.73/documente_shared/infrastructure/documente_client.py

@@ -0,0 +1,21 @@
+from requests import Session
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class DocumenteClientMixin(object):
+    api_url: str
+    api_key: str
+    session: Optional[Session] = None
+
+    def __post_init__(self):
+        self.session = Session()
+        self.session.headers.update(self.get_common_headers())
+
+
+    def get_common_headers(self) -> dict:
+        return {
+            "X-Api-Key": self.api_key,
+            "Content-Type": "application/json"
+        }

{documente_shared-0.1.72 → documente_shared-0.1.73}/documente_shared/infrastructure/repositories/dynamo_processing_case.py

@@ -13,8 +13,8 @@ class DynamoProcessingCaseRepository(
     DynamoDBTable,
     ProcessingCaseRepository,
 ):
-    def find(self, case_id: str) -> Optional[ProcessingCase]:
-        item = self.get(key={'case_id': case_id})
+    def find(self, uuid: str) -> Optional[ProcessingCase]:
+        item = self.get(key={'uuid': uuid})
         if item:
             return ProcessingCase.from_dict(item)
         return None
@@ -24,7 +24,7 @@ class DynamoProcessingCaseRepository(
         return instance
 
     def remove(self, instance: ProcessingCase):
-        self.delete(key={'case_id': instance.case_id})
+        self.delete(key={'uuid': instance.uuid})
 
     def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCase]:
         items = []

{documente_shared-0.1.72 → documente_shared-0.1.73}/documente_shared/infrastructure/repositories/dynamo_processing_case_item.py

@@ -13,8 +13,8 @@ class DynamoProcessingCaseItemRepository(
     DynamoDBTable,
     ProcessingCaseItemRepository,
 ):
-    def find(self, digest: str) -> Optional[ProcessingCaseItem]:
-        item = self.get(key={'digest': digest})
+    def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
+        item = self.get(key={'digest': uuid})
         if item:
             return ProcessingCaseItem.from_dict(item)
         return None

documente_shared-0.1.73/documente_shared/infrastructure/repositories/http_processing_case.py

@@ -0,0 +1,41 @@
+from dataclasses import dataclass
+from typing import List, Optional
+
+from documente_shared.domain.entities.processing_case import ProcessingCase
+from documente_shared.domain.enums.common import ProcessingStatus
+from documente_shared.domain.repositories.processing_case import ProcessingCaseRepository
+from documente_shared.infrastructure.documente_client import DocumenteClientMixin
+
+
+@dataclass
+class HttpProcessingCaseRepository(
+    DocumenteClientMixin,
+    ProcessingCaseRepository,
+):
+    def find(self, uuid: str) -> Optional[ProcessingCase]:
+        response = self.session.get(f"{self.api_url}/processing-cases/{uuid}/")
+        if response.status_code == 200:
+            return ProcessingCase.from_dict(response.json())
+        return None
+
+    def persist(self, instance: ProcessingCase) -> ProcessingCase:
+        response = self.session.put(
+            url=f"{self.api_url}/processing-cases/{instance.uuid}/",
+            json=instance.to_dict,
+        )
+        if response.status_code not in [200, 201]:
+            raise Exception(f'Error persisting processing case: {response.text}')
+        return ProcessingCase.from_dict(response.json())
+
+    def remove(self, instance: ProcessingCase):
+        self.session.delete(f"{self.api_url}/processing-cases/{instance.uuid}/")
+
+    def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCase]:
+        response = self.session.get(f"{self.api_url}/processing-cases/")
+        if response.status_code == 200:
+            raw_response = response.json()
+            return [
+                ProcessingCase.from_dict(item)
+                for item in raw_response.get('data', [])
+            ]
+        return []

documente_shared-0.1.73/documente_shared/infrastructure/repositories/http_processing_case_item.py

@@ -0,0 +1,53 @@
+from dataclasses import dataclass
+from typing import List, Optional
+
+from documente_shared.domain.entities.processing_case import ProcessingCase
+from documente_shared.domain.entities.processing_case_item import ProcessingCaseItem
+from documente_shared.domain.enums.common import ProcessingStatus
+from documente_shared.domain.repositories.processing_case_item import ProcessingCaseItemRepository
+from documente_shared.infrastructure.documente_client import DocumenteClientMixin
+
+
+@dataclass
+class HttpProcessingCaseItemRepository(
+    DocumenteClientMixin,
+    ProcessingCaseItemRepository,
+):
+
+    def find(self, uuid: str) -> Optional[ProcessingCaseItem]:
+        response = self.session.get(f"{self.api_url}/processing-case-items/{uuid}/")
+        if response.status_code == 200:
+            return ProcessingCaseItem.from_dict(response.json())
+        return None
+
+    def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
+        response = self.session.put(
+            url=f"{self.api_url}/processing-case-items/{instance.uuid}/",
+            json=instance.to_simple_dict,
+        )
+        if response.status_code in [200, 201]:
+            return ProcessingCaseItem.from_dict(response.json())
+        return instance
+
+    def remove(self, instance: ProcessingCaseItem):
+        self.session.delete(f"{self.api_url}/processing-case-items/{instance.uuid}/")
+
+    def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCase]:
+        response = self.session.get(f"{self.api_url}/processing-case-items/")
+        if response.status_code == 200:
+            raw_response = response.json()
+            return [
+                ProcessingCase.from_dict(item)
+                for item in raw_response.get('data', [])
+            ]
+        return []
+
+    def filter_by_case_id(self, case_id: str) -> List[ProcessingCase]:
+        response = self.session.get(f"{self.api_url}/processing-case-items/?case_id={case_id}")
+        if response.status_code == 200:
+            raw_response = response.json()
+            return [
+                ProcessingCase.from_dict(item)
+                for item in raw_response.get('data', [])
+            ]
+        return []

{documente_shared-0.1.72 → documente_shared-0.1.73}/pyproject.toml

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

documente_shared-0.1.72/documente_shared/domain/constants.py

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