documente_shared 0.1.70__tar.gz → 0.1.71__tar.gz

{documente_shared-0.1.70 → documente_shared-0.1.71}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.70
+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.70 → documente_shared-0.1.71}/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-0.1.70 → documente_shared-0.1.71}/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-0.1.70 → documente_shared-0.1.71}/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-0.1.70/documente_shared/domain/repositories.py → documente_shared-0.1.71/documente_shared/domain/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-0.1.71/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-0.1.71/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-0.1.70/documente_shared/infrastructure/dynamo_repositories.py → documente_shared-0.1.71/documente_shared/infrastructure/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-0.1.71/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-0.1.71/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-0.1.70 → documente_shared-0.1.71}/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 → documente_shared-0.1.71}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.70"
+version = "0.1.71"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"
@@ -8,9 +8,10 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
-boto3 = "^1.34.80"
-botocore = "^1.34.80"
+boto3 = "^1.37.19"
+botocore = "^1.37.19"
 sentry-sdk = "^2.19.2"
+pytz = "^2025.2"
 
 [tool.poetry.dev-dependencies]
 pytest = "^8.3.4"

{documente_shared-0.1.70 → documente_shared-0.1.71}/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(