documente_shared 0.1.80__py3-none-any.whl → 0.1.81__py3-none-any.whl

documente_shared/application/payloads.py

@@ -0,0 +1,14 @@
+import re
+
+
+def camel_to_snake_key(name: str) -> str:
+    s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
+    return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()
+
+def camel_to_snake(data: dict | list) -> dict | list:
+    if isinstance(data, dict):
+        return {camel_to_snake_key(k): camel_to_snake(v) for k, v in data.items()}
+    elif isinstance(data, list):
+        return [camel_to_snake(item) for item in data]
+    else:
+        return data

documente_shared/domain/entities/in_memory_result.py

@@ -11,6 +11,7 @@ from documente_shared.application.files import (
 class InMemoryDocument(object):
     file_path: Optional[str] = None
     file_bytes: Optional[bytes] = None
+    file_base64: Optional[str] = None
 
     @property
     def is_valid(self) -> bool:
@@ -18,13 +19,7 @@ class InMemoryDocument(object):
 
     @property
     def has_content(self) -> bool:
-        return bool(self.file_bytes)
-
-    @property
-    def file_key(self) -> Optional[str]:
-        if not self.file_path:
-            return None
-        return remove_slash_from_path(self.file_path)
+        return bool(self.file_bytes) or bool(self.file_base64)
 
     @property
     def file_name(self) -> Optional[str]:
@@ -32,20 +27,29 @@ class InMemoryDocument(object):
             return None
         return get_filename_from_path(self.file_path)
 
+    @property
+    def file_key(self) -> Optional[str]:
+        return self.file_name
+
     @property
     def is_procesable(self) -> bool:
         return self.is_valid and self.has_content
 
     @property
     def to_dict(self) -> dict:
-        return {
-            'file_path': self.file_path,
-            'file_bytes': self.file_bytes,
-        }
+        data = dict()
+        if self.file_path:
+            data['file_path'] = remove_slash_from_path(self.file_path)
+        if self.file_bytes:
+            data['file_bytes'] = self.file_bytes.decode('utf-8')
+        if self.file_base64:
+            data['file_base64'] = self.file_base64
+        return data
 
     @classmethod
     def from_dict(cls, data: dict):
         return cls(
             file_path=data.get('file_path'),
             file_bytes=data.get('file_bytes'),
+            file_base64=data.get('file_base64'),
         )

documente_shared/domain/entities/processing_case.py

@@ -96,37 +96,18 @@ class ProcessingCase(object):
 
     @property
     def to_persist_dict(self) -> dict:
-        return self.to_dict
-
-    def overload(
-        self,
-        new_instance: 'ProcessingCase',
-        properties: List[str] = None,
-    ):
-        instance_properties = properties or [
-            'label',
-            'status',
-            'category',
-            'enqueued_at',
-            'started_at',
-            'failed_at',
-            'feedback',
-            'completed_at',
-            'metadata',
-            'items',
+        persist_data = self.to_dict
+        persist_data["items"] = [
+            item.to_persist_dict for item in self.items
         ]
-        for _property in instance_properties:
-            property_value = getattr(new_instance, _property)
-            if not hasattr(self, _property):
-                continue
-            setattr(self, _property, property_value)
-        return self
+        return persist_data
 
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingCase':
         return cls(
             uuid=data.get('uuid'),
             name=data.get('label'),
+            tenant_slug=data.get('tenant_slug'),
             status=ProcessingStatus.from_value(data.get('status')),
             case_type=(
                 ProcessingCaseType.from_value(data.get('category'))
@@ -143,3 +124,12 @@ class ProcessingCase(object):
                 for item_dict in data.get('items', [])
             ],
         )
+
+    @classmethod
+    def from_persist_dict(cls, data: dict) -> 'ProcessingCase':
+        instance = cls.from_dict(data)
+        instance.items = [
+            ProcessingCaseItem.from_persist_dict(item_dict)
+            for item_dict in data.get('items', [])
+        ]
+        return instance

documente_shared/domain/entities/processing_case_item.py

@@ -205,7 +205,10 @@ class ProcessingCaseItem(object):
             case_id=data.get('case_id'),
             digest=data.get('digest'),
             status=ProcessingStatus.from_value(data.get('status')),
-            document=InMemoryDocument.from_dict(data.get('document')),
+            document=(
+                InMemoryDocument.from_dict(data.get('document'))
+                if data.get('document') else None
+            ),
             document_type=(
                 ProcessingDocumentType.from_value(data.get('document_type'))
                 if data.get('document_type') else None

documente_shared/infrastructure/repositories/http_processing_case.py

@@ -1,6 +1,7 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
+from documente_shared.application.payloads import camel_to_snake
 from documente_shared.domain.entities.processing_case import ProcessingCase
 from documente_shared.domain.entities.processing_case_filters import ProcessingCaseFilters
 from documente_shared.domain.repositories.processing_case import ProcessingCaseRepository
@@ -15,7 +16,9 @@ class HttpProcessingCaseRepository(
     def find(self, uuid: str, include_items: bool = False) -> Optional[ProcessingCase]:
         response = self.session.get(f"{self.api_url}/v1/processing-cases/{uuid}/")
         if response.status_code == 200:
-            return ProcessingCase.from_dict(response.json())
+            response_json = response.json()
+            instance_data = response_json.get('data', {})
+            return ProcessingCase.from_persist_dict(camel_to_snake(instance_data))
         return None
 
     def persist(self, instance: ProcessingCase) -> ProcessingCase:
@@ -25,7 +28,10 @@ class HttpProcessingCaseRepository(
         )
         if response.status_code not in [200, 201]:
             raise Exception(f'Error persisting processing case: {response.text}')
-        return ProcessingCase.from_dict(response.json())
+
+        response_json = response.json()
+        instance_data = response_json.get('data', {})
+        return ProcessingCase.from_persist_dict(camel_to_snake(instance_data))
 
     def remove(self, instance: ProcessingCase):
         self.session.delete(f"{self.api_url}/v1/processing-cases/{instance.uuid}/")
@@ -39,8 +45,9 @@ class HttpProcessingCaseRepository(
         )
         if response.status_code == 200:
             raw_response = response.json()
+            instaces_data = raw_response.get('data', [])
             return [
-                ProcessingCase.from_dict(item)
-                for item in raw_response.get('data', [])
+                ProcessingCase.from_persist_dict(item_data)
+                for item_data in camel_to_snake(instaces_data)
             ]
         return []

{documente_shared-0.1.80.dist-info → documente_shared-0.1.81.dist-info}/METADATA

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

{documente_shared-0.1.80.dist-info → documente_shared-0.1.81.dist-info}/RECORD

@@ -3,6 +3,7 @@ documente_shared/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 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
+documente_shared/application/payloads.py,sha256=s6SjaNN18_aQ6IL083Zq2J8thRCZ_zC2sn7hkfjK_Go,453
 documente_shared/application/query_params.py,sha256=JscPqFBx28p-x9i2g6waY7Yl4FQM1zn2zSbEoTrkK1k,3938
 documente_shared/application/time_utils.py,sha256=_fxgh8VoGPkdsft47COJ16vFwt8pMbHIJCgDFHLSlrU,435
 documente_shared/application/timezone.py,sha256=NHpzTzOPD_fWQiJ4BrRqt_TIDs5XyB5ZMR7x8vUk8gQ,183
@@ -12,10 +13,10 @@ documente_shared/domain/constants.py,sha256=NG5BGaXBr_FnzudjTRPxpDpyiSDdaB_PLCdl
 documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/entities/document.py,sha256=AthTUyA-QZE3WAT7lMoKVr_Z8mO_3qERuCnZge0DTLQ,12595
 documente_shared/domain/entities/document_metadata.py,sha256=ygyFIC5qwxlm8DUM5kvVFny9zJfPQS8vNLM2br5XsQ8,2353
-documente_shared/domain/entities/in_memory_result.py,sha256=Q1E9vnLL5Hz5xunOqWtQmJOMjoK5KN42LZr18GlBAZo,1246
-documente_shared/domain/entities/processing_case.py,sha256=3bL6BgFwWe6F5keZ6A1K_lLsrwGpkcKg98roM_i6kEQ,5167
+documente_shared/domain/entities/in_memory_result.py,sha256=0sLNUrovKFQx4M-E9e4DrAiVgch2i4AKA-9BQBRaeI8,1482
+documente_shared/domain/entities/processing_case.py,sha256=59fhosBfKh9ijWRDTkdN9agV3Qa_AIwjHEWU6FBPF4k,4978
 documente_shared/domain/entities/processing_case_filters.py,sha256=FgyxB4mQb0nEGjIbUB9OiazkKL4yHRRC6bvmjD5NT8k,1915
-documente_shared/domain/entities/processing_case_item.py,sha256=zav9tZ3EjQIJMccQzguF_0UJY4vc3yzfcNEQI_evMj4,9578
+documente_shared/domain/entities/processing_case_item.py,sha256=5BqKAv56R3tba0_rPdYtiTCpJ9ytP-JEjkCZgr1cjh0,9660
 documente_shared/domain/entities/processing_case_item_filters.py,sha256=-cAQTSWOepMMcGCBg2X3dd0W_8XHuBTlvOB1d-3sVVM,1971
 documente_shared/domain/entities/processing_event.py,sha256=m1O0gcNaE_SszeIhxM3uYPHSpyOUmize6mfRw1_bYZo,1723
 documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -33,10 +34,10 @@ documente_shared/infrastructure/repositories/__init__.py,sha256=47DEQpj8HBSa-_TI
 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/http_processing_case.py,sha256=-zNGiFCfINFWgE2ISWcEvPZThZCp45WRQhKFEL7QZW8,1869
+documente_shared/infrastructure/repositories/http_processing_case.py,sha256=zeGkuvF1SZMyDOUunp93fhn7lS61NQKa1GBcNF8dzZU,2251
 documente_shared/infrastructure/repositories/http_processing_case_item.py,sha256=OSkOd7WdksS4O75uALU530kCsLkpjRTEBCmg3wg0ZFY,2110
 documente_shared/infrastructure/s3_bucket.py,sha256=vT_yN42RFQXubtUn8ln-j13Os_-25UGClVtXg5Bkv6I,1932
 documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
-documente_shared-0.1.80.dist-info/METADATA,sha256=2RZnPEBEtepeyxgb9XCEQ8jbrItRuEyhQf6MvAfbOgY,881
-documente_shared-0.1.80.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.80.dist-info/RECORD,,
+documente_shared-0.1.81.dist-info/METADATA,sha256=cPO-JJd3KwUjCci_9-5KKWqrDSokT-x2VCNZpoBW_Rk,881
+documente_shared-0.1.81.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.81.dist-info/RECORD,,