documente_shared 0.1.70__py3-none-any.whl → 0.1.72__py3-none-any.whl

documente_shared/domain/entities/document.py

@@ -4,19 +4,18 @@ from datetime import datetime, tzinfo
 from decimal import Decimal
 from typing import Optional, List
 
-from documente_shared.application.files import remove_slash_from_path, split_file_params, get_filename_from_path
+from documente_shared.application.files import remove_slash_from_path, get_filename_from_path
 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.document_metadata import DocumentProcessingMetadata
-from documente_shared.domain.enums import (
+from documente_shared.domain.enums.document import (
     DocumentProcessingStatus,
+    DocumentProcessingCategory,
     DocumentProcessingSubCategory,
-    DocumentProcessingCategory, DocumentProcessingSource,
+    DocumentProcessingSource,
 )
 
 
-
-
 @dataclass
 class DocumentProcessing(object):
     digest: str

documente_shared/domain/entities/processing_case.py

@@ -11,7 +11,8 @@ from documente_shared.domain.enums.processing_case import ProcessingCaseCategory
 
 @dataclass
 class ProcessingCase(object):
-    id: str
+    uuid: str
+    label: str
     status: ProcessingStatus
     category: Optional[ProcessingCaseCategory] = None
     enqueued_at: Optional[datetime] = None
@@ -61,21 +62,23 @@ class ProcessingCase(object):
             return False
 
         return (
-            self.id == other.id
-            and self.status == other.status
-            and self.category == other.category
-            and self.enqueued_at == other.enqueued_at
-            and self.started_at == other.started_at
-            and self.failed_at == other.failed_at
-            and self.feedback == other.feedback
-            and self.completed_at == other.completed_at
-            and self.metadata == other.metadata
+                self.uuid == other.uuid
+                and self.label == other.label
+                and self.status == other.status
+                and self.category == other.category
+                and self.enqueued_at == other.enqueued_at
+                and self.started_at == other.started_at
+                and self.failed_at == other.failed_at
+                and self.feedback == other.feedback
+                and self.completed_at == other.completed_at
+                and self.metadata == other.metadata
         )
 
     @property
     def to_dict(self) -> dict:
         return {
-            'id': self.id,
+            'uuid': self.uuid,
+            'label': self.label,
             'status': str(self.status),
             'category': (
                 str(self.category)
@@ -100,6 +103,7 @@ class ProcessingCase(object):
         properties: List[str] = None,
     ):
         instance_properties = properties or [
+            'label',
             'status',
             'category',
             'enqueued_at',
@@ -120,7 +124,8 @@ class ProcessingCase(object):
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingCase':
         return cls(
-            id=data.get('id'),
+            uuid=data.get('uuid'),
+            label=data.get('label'),
             status=ProcessingStatus.from_value(data.get('status')),
             category=(
                 ProcessingCaseCategory.from_value(data.get('category'))

documente_shared/domain/entities/processing_case_item.py

@@ -12,7 +12,7 @@ from documente_shared.domain.enums.processing_case import ProcessingDocumentType
 
 @dataclass
 class ProcessingCaseItem(object):
-    id: str
+    uuid: str
     case_id: str
     digest: str
     status: ProcessingStatus
@@ -24,6 +24,7 @@ class ProcessingCaseItem(object):
     processed_json: Optional[InMemoryDocument] = None
     processing_time: Optional[Decimal] = None
     processing_confidence: Optional[Decimal] = None
+    uploaded_at: Optional[datetime] = None
     started_at: Optional[datetime] = None
     failed_at: Optional[datetime] = None
     completed_at: Optional[datetime] = None
@@ -85,12 +86,14 @@ class ProcessingCaseItem(object):
             return False
 
         return (
-            self.digest == other.digest
+            self.uuid == other.uuid
+            and self.digest == other.digest
             and self.status == other.status
             and self.document_type == other.document_type
             and self.document == other.document
             and self.processing_time == other.processing_time
             and self.processing_confidence == other.processing_confidence
+            and self.uploaded_at == other.uploaded_at
             and self.started_at == other.started_at
             and self.failed_at == other.failed_at
             and self.completed_at == other.completed_at
@@ -99,12 +102,12 @@ class ProcessingCaseItem(object):
     @property
     def to_dict(self) -> dict:
         return {
-            'id': self.id,
+            'uuid': self.uuid,
             'case_id': self.case_id,
             'digest': self.digest,
             'status': str(self.status),
             'document': self.document.to_dict,
-            'document_type': self.document_type,
+            'document_type': str(self.document_type),
             'uploaded_from': (
                 str(self.uploaded_from)
                 if self.uploaded_from else None
@@ -129,6 +132,7 @@ class ProcessingCaseItem(object):
                 str(self.processing_confidence.quantize(Decimal('0.001')))
                 if self.processing_confidence else None
             ),
+            'uploaded_at': self.uploaded_at.isoformat() if self.uploaded_at else None,
             'started_at': self.started_at.isoformat() if self.started_at else None,
             'failed_at': self.failed_at.isoformat() if self.failed_at else None,
             'feedback': self.feedback,
@@ -139,7 +143,6 @@ class ProcessingCaseItem(object):
     @property
     def to_simple_dict(self) -> dict:
         simple_dict = self.to_dict.copy()
-        simple_dict.pop('metadata_items')
         return simple_dict
 
     def overload(
@@ -157,6 +160,7 @@ class ProcessingCaseItem(object):
             'processed_json',
             'processing_time',
             'processing_confidence',
+            'uploaded_at',
             'started_at',
             'failed_at',
             'completed_at',
@@ -173,7 +177,7 @@ class ProcessingCaseItem(object):
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingCaseItem':
         return cls(
-            id=data.get('id'),
+            uuid=data.get('uuid'),
             case_id=data.get('case_id'),
             digest=data.get('digest'),
             status=ProcessingStatus.from_value(data.get('status')),
@@ -203,6 +207,7 @@ class ProcessingCaseItem(object):
                 Decimal(data.get('processing_confidence'))
                 if data.get('processing_confidence') else None
             ),
+            uploaded_at=get_datetime_from_data(input_datetime=data.get('uploaded_at')),
             started_at=get_datetime_from_data(input_datetime=data.get('started_at')),
             failed_at=get_datetime_from_data(input_datetime=data.get('failed_at')),
             feedback=data.get('feedback'),

documente_shared/domain/{repositories.py → repositories/document.py}

@@ -2,13 +2,13 @@ from abc import ABC, abstractmethod
 from typing import Optional, List
 
 from documente_shared.domain.entities.document import DocumentProcessing
-from documente_shared.domain.enums import DocumentProcessingStatus
+from documente_shared.domain.enums.document import DocumentProcessingStatus
 
 
 class DocumentProcessingRepository(ABC):
 
     @abstractmethod
-    def find(self, digest: str) ->Optional[DocumentProcessing]:
+    def find(self, digest: str) -> Optional[DocumentProcessing]:
         raise NotImplementedError
 
     @abstractmethod
@@ -19,7 +19,6 @@ class DocumentProcessingRepository(ABC):
     def remove(self, instance: DocumentProcessing):
         raise NotImplementedError
 
-
     @abstractmethod
     def filter(self, statuses: List[DocumentProcessingStatus]) -> List[DocumentProcessing]:
-        raise NotImplementedError
+        raise NotImplementedError

documente_shared/domain/repositories/processing_case.py

@@ -0,0 +1,24 @@
+from abc import ABC, abstractmethod
+from typing import Optional, List
+
+from documente_shared.domain.entities.processing_case import ProcessingCase
+from documente_shared.domain.enums.common import ProcessingStatus
+
+
+class ProcessingCaseRepository(ABC):
+
+    @abstractmethod
+    def find(self, case_id: str) -> Optional[ProcessingCase]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def persist(self, instance: ProcessingCase) -> ProcessingCase:
+        raise NotImplementedError
+
+    @abstractmethod
+    def remove(self, instance: ProcessingCase):
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCase]:
+        raise NotImplementedError

documente_shared/domain/repositories/processing_case_item.py

@@ -0,0 +1,29 @@
+from abc import ABC, abstractmethod
+from typing import Optional, List
+
+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
+
+
+class ProcessingCaseItemRepository(ABC):
+
+    @abstractmethod
+    def find(self, digest: str) -> Optional[ProcessingCaseItem]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
+        raise NotImplementedError
+
+    @abstractmethod
+    def remove(self, instance: ProcessingCaseItem):
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCase]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter_by_case_id(self, case_id: str) -> List[ProcessingCase]:
+        raise NotImplementedError

documente_shared/infrastructure/dynamo_table.py

@@ -1,11 +1,12 @@
-from dataclasses import dataclass
-
 import boto3
+
+from dataclasses import dataclass
 from boto3.dynamodb.conditions import Key
 
 
 RETURN_VALUES = 'UPDATED_NEW'
 
+
 @dataclass
 class DynamoDBTable(object):
     table_name: str
@@ -50,7 +51,6 @@ class DynamoDBTable(object):
     def count(self) -> int:
         return self._table.item_count
 
-
     @classmethod
     def _update_expression(cls, attributes):
         return 'SET {param}'.format(

documente_shared/infrastructure/{dynamo_repositories.py → repositories/dynamo_document.py}

@@ -3,10 +3,10 @@ from typing import Optional, List
 from boto3.dynamodb.conditions import Key
 
 from documente_shared.domain.entities.document import DocumentProcessing
-from documente_shared.domain.enums import DocumentProcessingStatus, DocumentProcessingCategory
-from documente_shared.domain.repositories import DocumentProcessingRepository
-from documente_shared.infrastructure.dynamo_table import DynamoDBTable
+from documente_shared.domain.enums.document import DocumentProcessingStatus
+from documente_shared.domain.repositories.document import DocumentProcessingRepository
 
+from documente_shared.infrastructure.dynamo_table import DynamoDBTable
 
 
 class DynamoDocumentProcessingRepository(

documente_shared/infrastructure/repositories/dynamo_processing_case.py

@@ -0,0 +1,43 @@
+from typing import Optional, List
+
+from boto3.dynamodb.conditions import Key
+
+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.dynamo_table import DynamoDBTable
+
+
+class DynamoProcessingCaseRepository(
+    DynamoDBTable,
+    ProcessingCaseRepository,
+):
+    def find(self, case_id: str) -> Optional[ProcessingCase]:
+        item = self.get(key={'case_id': case_id})
+        if item:
+            return ProcessingCase.from_dict(item)
+        return None
+
+    def persist(self, instance: ProcessingCase) -> ProcessingCase:
+        self.put(instance.to_persist_dict)
+        return instance
+
+    def remove(self, instance: ProcessingCase):
+        self.delete(key={'case_id': instance.case_id})
+
+    def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCase]:
+        items = []
+
+        for status in statuses:
+            response = self._table.query(
+                IndexName='status',
+                KeyConditionExpression=Key('status').eq(status.value),
+            )
+            status_items = response.get('Items', [])
+            items.extend(status_items)
+
+        return [
+            ProcessingCase.from_dict(item)
+            for item in items
+        ]

documente_shared/infrastructure/repositories/dynamo_processing_case_item.py

@@ -0,0 +1,54 @@
+from typing import Optional, List
+
+from boto3.dynamodb.conditions import Key
+
+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.dynamo_table import DynamoDBTable
+
+
+class DynamoProcessingCaseItemRepository(
+    DynamoDBTable,
+    ProcessingCaseItemRepository,
+):
+    def find(self, digest: str) -> Optional[ProcessingCaseItem]:
+        item = self.get(key={'digest': digest})
+        if item:
+            return ProcessingCaseItem.from_dict(item)
+        return None
+
+    def persist(self, instance: ProcessingCaseItem) -> ProcessingCaseItem:
+        self.put(instance.to_simple_dict)
+        return instance
+
+    def remove(self, instance: ProcessingCaseItem):
+        self.delete(key={'case_id': instance.case_id})
+
+    def filter(self, statuses: List[ProcessingStatus]) -> List[ProcessingCaseItem]:
+        items = []
+
+        for status in statuses:
+            response = self._table.query(
+                IndexName='status',
+                KeyConditionExpression=Key('status').eq(status.value),
+            )
+            status_items = response.get('Items', [])
+            items.extend(status_items)
+
+        return [
+            ProcessingCaseItem.from_dict(item)
+            for item in items
+        ]
+
+    def filter_by_case_id(self, case_id: str) -> List[ProcessingCaseItem]:
+        response = self._table.query(
+            IndexName='case_id',
+            KeyConditionExpression=Key('case_id').eq(case_id),
+        )
+        items = response.get('Items', [])
+        return [
+            ProcessingCaseItem.from_dict(item)
+            for item in items
+        ]

documente_shared/infrastructure/sqs_queue.py

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 @dataclass
 class SQSQueue(object):
     queue_url: str
-    visibility_timeout: int =  60 * 10
+    visibility_timeout: int = 60 * 10
     waiting_timeout: int = 20
 
     def __post_init__(self):

{documente_shared-0.1.70.dist-info → documente_shared-0.1.72.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.70
+Version: 0.1.72
 Summary: Shared utilities for Documente AI projects
 License: MIT
 Author: Tech
@@ -11,8 +11,9 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: boto3 (>=1.34.80,<2.0.0)
-Requires-Dist: botocore (>=1.34.80,<2.0.0)
+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: sentry-sdk (>=2.19.2,<3.0.0)
 Description-Content-Type: text/markdown
 

{documente_shared-0.1.70.dist-info → documente_shared-0.1.72.dist-info}/RECORD

@@ -9,22 +9,28 @@ documente_shared/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 documente_shared/domain/base_enum.py,sha256=DojAfn-zQdtjtImeHUpBzE6TBTm07XrbMOdW3h8RVd8,1449
 documente_shared/domain/constants.py,sha256=jOlMKFq12FgiYMJcQHku8IVwuOE5t-HEPuSV_zEeIFo,56
 documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/entities/document.py,sha256=kaLuVJ-LVQMUdQhCReQ2Mkl7-Kmws6-TNxKnlXogYSk,12603
+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=PuwWQuXIIP5wlkWBCrkF-61TfgKSYwc9HZl1JR13xhk,4955
-documente_shared/domain/entities/processing_case_item.py,sha256=5wuZylIc_Lanh5PnY4ghEgbQhCtBDCExh4KRW6ZGK_4,7684
+documente_shared/domain/entities/processing_case.py,sha256=rGbFs3XhKppGOfH3XMa2BiCyr7lnqRMjhDXM6oUwKXE,5157
+documente_shared/domain/entities/processing_case_item.py,sha256=hglyPhZBSjlMKVdEVccyHXU3aQRdZ8hfZlXDJnkf7wA,7996
 documente_shared/domain/entities/processing_event.py,sha256=m1O0gcNaE_SszeIhxM3uYPHSpyOUmize6mfRw1_bYZo,1723
 documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/enums/common.py,sha256=vXldMUPhhWo0PfTgYwDSjI8bur_lYcImZYiV7yAO7DQ,2262
 documente_shared/domain/enums/document.py,sha256=NltZA1YVgJ7dVfSQdJFIE0ZUGf9Y-nxNXsVQ6GiPLL4,1827
 documente_shared/domain/enums/processing_case.py,sha256=DLVk0VnVzrKs1CvXVGHV9p8mBUHLUrOSJJ1POE-Ek3o,1546
-documente_shared/domain/repositories.py,sha256=g3qLUy2kT8esmvU4VxxSVnDaXeySKKQ7mUvIvxOwh9A,757
+documente_shared/domain/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/domain/repositories/document.py,sha256=vJzr6c92gqBzyhaEdjrvnoneKRrWmJ0AsvocPnhxiLU,767
+documente_shared/domain/repositories/processing_case.py,sha256=7WE5RdLr04ysGXAdkJYmbfcLpMyOMeuDJGounZKMZqs,729
+documente_shared/domain/repositories/processing_case_item.py,sha256=yD-v_24UzHwNdq75KRJGwHemTVFpO9LMIPSup4GbyKs,959
 documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/infrastructure/dynamo_repositories.py,sha256=SEad_HLppp2h_BKDSzb9oo1VlAVRZWelOPvJPlDwbzQ,1453
-documente_shared/infrastructure/dynamo_table.py,sha256=dK05KgFvIYCmOdMpq9-OV_OBrP6cCngiUikCJrxlwt4,2112
+documente_shared/infrastructure/dynamo_table.py,sha256=TMQbcuty7wjDMbuhI8PbT0IGXelgELsNTtqTEQeZ824,2112
+documente_shared/infrastructure/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/infrastructure/repositories/dynamo_document.py,sha256=_Yp4gtA-n-hJ2w2wAM5BMCs2Mf46Q2Kq3eHqlxudkL4,1443
+documente_shared/infrastructure/repositories/dynamo_processing_case.py,sha256=Wt0di0402mrnvYNqe5ywDtFFdXjnHHvLDG4k_MQfmTE,1401
+documente_shared/infrastructure/repositories/dynamo_processing_case_item.py,sha256=gG7H7hyeKTPMywTUmnq0nClrpaOqVPy7bh1OLBzV4pE,1816
 documente_shared/infrastructure/s3_bucket.py,sha256=vT_yN42RFQXubtUn8ln-j13Os_-25UGClVtXg5Bkv6I,1932
-documente_shared/infrastructure/sqs_queue.py,sha256=PSiTAnjXvQ-W-9mzLpH2UjbQJTvYkMiaxNaMecF-cR4,1505
-documente_shared-0.1.70.dist-info/METADATA,sha256=JCm5S835mggDqZMKE320WqYbKgyH79nvdZMXIIaMnEU,800
-documente_shared-0.1.70.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.70.dist-info/RECORD,,
+documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
+documente_shared-0.1.72.dist-info/METADATA,sha256=k9oTLxK2WigrpoZx6Psxtqs_OOBWWM-6VuAXZ_kjUVc,839
+documente_shared-0.1.72.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.72.dist-info/RECORD,,