documente_shared 0.1.14__tar.gz → 0.1.16__tar.gz

{documente_shared-0.1.14 → documente_shared-0.1.16}/PKG-INFO

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

documente_shared-0.1.16/documente_shared/application/digest.py

@@ -0,0 +1,7 @@
+import hashlib
+
+
+def get_file_digest(file_bytes: bytes) -> str:
+     sha256_hash = hashlib.sha256()
+     sha256_hash.update(file_bytes)
+     return sha256_hash.hexdigest()

{documente_shared-0.1.14 → documente_shared-0.1.16}/documente_shared/domain/entities.py

@@ -3,7 +3,11 @@ from datetime import datetime
 from decimal import Decimal
 from typing import Optional
 
-from documente_shared.domain.enums import DocumentProcessStatus
+from documente_shared.domain.enums import (
+    DocumentProcessStatus,
+    DocumentProcessSubCategory,
+    DocumentProcessCategory,
+)
 
 
 @dataclass
@@ -11,13 +15,16 @@ class DocumentProcess(object):
     digest: str
     status: DocumentProcessStatus
     file_path: str
+    category: Optional[DocumentProcessCategory] = None
+    sub_category: Optional[DocumentProcessSubCategory] = None
     processed_csv_path: Optional[str] = None
     processed_xlsx_path: Optional[str] = None
     processing_time: Optional[Decimal] = None
+    uploded_at: Optional[datetime] = None
     enqueued_at: Optional[datetime] = None
     started_at: Optional[datetime] = None
     failed_at: Optional[datetime] = None
-    processed_at: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
 
     @property
     def is_pending(self) -> bool:
@@ -61,7 +68,7 @@ class DocumentProcess(object):
 
     def completed(self):
         self.status = DocumentProcessStatus.COMPLETED
-        self.processed_at = datetime.now()
+        self.completed_at = datetime.now()
 
     def deleted(self):
         self.status = DocumentProcessStatus.DELETED
@@ -77,16 +84,25 @@ class DocumentProcess(object):
             'digest': self.digest,
             'status': self.status.value,
             'file_path': self.file_path,
+            'category': (
+                str(self.category)
+                if self.category else None
+            ),
+            'sub_category': (
+                str(self.sub_category)
+                if self.sub_category else None
+            ),
             'processed_csv_path': self.processed_csv_path,
             'processed_xlsx_path': self.processed_xlsx_path,
             'processing_time': (
                 str(self.processing_time)
                 if self.processing_time else None
             ),
+            'uploded_at': self.uploded_at.isoformat() if self.uploded_at else None,
             'enqueued_at': self.enqueued_at.isoformat() if self.enqueued_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,
-            'processed_at': self.processed_at.isoformat() if self.processed_at else None,
+            'completed_at': self.completed_at.isoformat() if self.completed_at else None,
         }
 
     @classmethod
@@ -95,12 +111,24 @@ class DocumentProcess(object):
             digest=data.get('digest'),
             status=DocumentProcessStatus.from_value(data.get('status')),
             file_path=data.get('file_path'),
+            category=(
+                DocumentProcessCategory.from_value(data.get('category'))
+                if data.get('category') else None
+            ),
+            sub_category=(
+                DocumentProcessSubCategory.from_value(data.get('sub_category'))
+                if data.get('sub_category') else None
+            ),
             processed_csv_path=data.get('processed_csv_path'),
             processed_xlsx_path=data.get('processed_xlsx_path'),
             processing_time=(
                 Decimal(data.get('processing_time'))
                 if data.get('processing_time') else None
             ),
+            uploded_at=(
+                datetime.fromisoformat(data.get('uploded_at'))
+                if data.get('uploded_at') else None
+            ),
             enqueued_at=(
                 datetime.fromisoformat(data.get('enqueued_at'))
                 if data.get('enqueued_at') else None
@@ -113,8 +141,8 @@ class DocumentProcess(object):
                 datetime.fromisoformat(data.get('failed_at'))
                 if data.get('failed_at') else None
             ),
-            processed_at=(
-                datetime.fromisoformat(data.get('processed_at'))
+            completed_at=(
+                datetime.fromisoformat(data.get('completed_at'))
                 if data.get('processed_at') else None
             ),
         )

{documente_shared-0.1.14 → documente_shared-0.1.16}/documente_shared/domain/enums.py

@@ -9,3 +9,13 @@ class DocumentProcessStatus(BaseEnum):
     FAILED = 'FAILED'
     DELETED = 'DELETED'
 
+
+class DocumentProcessCategory(BaseEnum):
+    CIRCULAR = 'CIRCULAR'
+
+
+class DocumentProcessSubCategory(BaseEnum):
+    CC_COMBINADA = 'CC_COMBINADA'
+    CC_NORMATIVA = 'CC_NORMATIVA'
+
+

{documente_shared-0.1.14 → documente_shared-0.1.16}/documente_shared/domain/repositories.py

@@ -1,10 +1,15 @@
 from abc import ABC, abstractmethod
 
 from documente_shared.domain.entities import DocumentProcess
+from python_layer.python.typing_extensions import Optional
 
 
 class DocumentProcessRepository(ABC):
 
+    @abstractmethod
+    def find(self, digest: str) ->Optional[DocumentProcess]:
+        raise NotImplementedError
+
     @abstractmethod
     def persist(self, instance: DocumentProcess) -> DocumentProcess:
         raise NotImplementedError

{documente_shared-0.1.14 → documente_shared-0.1.16}/documente_shared/infrastructure/dynamo_repositories.py

@@ -1,12 +1,18 @@
 from documente_shared.domain.entities import DocumentProcess
 from documente_shared.domain.repositories import DocumentProcessRepository
 from documente_shared.infrastructure.dynamo_table import DynamoDBTable
+from python_layer.python.typing_extensions import Optional
 
 
 class DynamoDocumentProcessRepository(
     DynamoDBTable,
     DocumentProcessRepository,
 ):
+    def find(self, digest: str) -> Optional[DocumentProcess]:
+        item = self.get(key=digest)
+        if item:
+            return DocumentProcess.from_dict(item)
+        return None
 
     def persist(self, instance: DocumentProcess) -> DocumentProcess:
         self.put(instance.to_dict)

{documente_shared-0.1.14 → documente_shared-0.1.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.14"
+version = "0.1.16"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"