documente_shared 0.1.83__tar.gz → 0.1.85__tar.gz

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

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

{documente_shared-0.1.83 → documente_shared-0.1.85}/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

{documente_shared-0.1.83 → documente_shared-0.1.85}/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.83 → documente_shared-0.1.85}/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.85/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.83 → documente_shared-0.1.85}/pyproject.toml

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