documente_shared 0.1.84__tar.gz → 0.1.86__tar.gz

{documente_shared-0.1.84 → documente_shared-0.1.86}/PKG-INFO

@@ -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.86/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.84 → documente_shared-0.1.86}/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-0.1.84 → documente_shared-0.1.86}/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-0.1.84 → documente_shared-0.1.86}/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-0.1.86/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.86/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-0.1.86/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 → documente_shared-0.1.86}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.84"
+version = "0.1.86"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"