documente_shared 0.1.70__tar.gz → 0.1.72__tar.gz

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

@@ -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 → documente_shared-0.1.72}/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.72}/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-0.1.70 → documente_shared-0.1.72}/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-0.1.70/documente_shared/domain/repositories.py → documente_shared-0.1.72/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.72/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.72/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.72/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.72/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.72/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.72}/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.72}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.70"
+version = "0.1.72"
 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.72}/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(