documente_shared 0.1.68__py3-none-any.whl → 0.1.71__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,7 @@ from documente_shared.domain.enums.processing_case import ProcessingCaseCategory
 
 @dataclass
 class ProcessingCase(object):
-    id: str
+    case_id: str
     status: ProcessingStatus
     category: Optional[ProcessingCaseCategory] = None
     enqueued_at: Optional[datetime] = None
@@ -61,21 +61,21 @@ 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.case_id == other.case_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
         )
 
     @property
     def to_dict(self) -> dict:
         return {
-            'id': self.id,
+            'case_id': self.case_id,
             'status': str(self.status),
             'category': (
                 str(self.category)
@@ -120,7 +120,7 @@ class ProcessingCase(object):
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingCase':
         return cls(
-            id=data.get('id'),
+            case_id=data.get('case_id'),
             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
+    item_id: str
     case_id: str
     digest: str
     status: ProcessingStatus
@@ -99,12 +99,12 @@ class ProcessingCaseItem(object):
     @property
     def to_dict(self) -> dict:
         return {
-            'id': self.id,
+            'item_id': self.item_id,
             '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
@@ -139,7 +139,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(
@@ -173,7 +172,7 @@ class ProcessingCaseItem(object):
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingCaseItem':
         return cls(
-            id=data.get('id'),
+            item_id=data.get('item_id'),
             case_id=data.get('case_id'),
             digest=data.get('digest'),
             status=ProcessingStatus.from_value(data.get('status')),

documente_shared/domain/entities/processing_event.py

@@ -0,0 +1,49 @@
+from dataclasses import dataclass
+from datetime import datetime
+from typing import Optional
+
+from documente_shared.application.time_utils import get_datetime_from_data
+from documente_shared.domain.entities.document import DocumentProcessing
+from documente_shared.domain.entities.processing_case import ProcessingCase
+from documente_shared.domain.enums.common import ProcessingType
+
+
+@dataclass
+class ProcessingEvent(object):
+    processing_type: ProcessingType
+    instance: DocumentProcessing | ProcessingCase | None
+    timestamp: Optional[datetime] = None
+
+    def __eq__(self, other: 'ProcessingEvent') -> bool:
+        if not other:
+            return False
+
+        return (
+            self.processing_type == other.processing_type
+            and self.instance == other.instance
+        )
+
+    @property
+    def to_dict(self) -> dict:
+        return {
+            'processing_type': str(self.processing_type),
+            'instance': self.instance.to_dict,
+            'timestamp': self.timestamp.isoformat() if self.timestamp else None,
+        }
+
+    @classmethod
+    def from_dict(cls, data: dict) -> 'ProcessingEvent':
+        processing_type = ProcessingType.from_value(data.get('processing_type'))
+
+        if processing_type.is_document:
+            processing_instance = DocumentProcessing.from_dict(data.get('instance'))
+        elif processing_type.is_processing_case:
+            processing_instance = ProcessingCase.from_dict(data.get('instance'))
+        else:
+            processing_instance = None
+
+        return cls(
+            processing_type=processing_type,
+            instance=processing_instance,
+            timestamp=get_datetime_from_data(input_datetime=data.get('timestamp')),
+        )

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.68.dist-info → documente_shared-0.1.71.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.68
+Version: 0.1.71
 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.68.dist-info → documente_shared-0.1.71.dist-info}/RECORD

@@ -9,21 +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=xIU10DSkh_7_1DU_ttx02eS020FOtCL64dp0kf0Hgm8,5026
+documente_shared/domain/entities/processing_case_item.py,sha256=rC8tE2qSw8SaOneKrg_cHtdD2WIAvFYht2qwxQdBuBU,7672
+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.68.dist-info/METADATA,sha256=GzKC9St8yRT3mlxTKnKVBx7xbfwrNTmsX239ZvmARV0,800
-documente_shared-0.1.68.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.68.dist-info/RECORD,,
+documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
+documente_shared-0.1.71.dist-info/METADATA,sha256=3TPrU9cuz9uYXEg7_sJer4h_bRt6p19Zmx1WHCbxbSw,839
+documente_shared-0.1.71.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.71.dist-info/RECORD,,