documente_shared 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

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,19 @@ class DocumentProcess(object):
     digest: str
     status: DocumentProcessStatus
     file_path: str
+    file_bytes: Optional[bytes] = None
+    category: Optional[DocumentProcessCategory] = None
+    sub_category: Optional[DocumentProcessSubCategory] = None
     processed_csv_path: Optional[str] = None
+    processed_csv_bytes: Optional[bytes] = None
     processed_xlsx_path: Optional[str] = None
+    processed_xlsx_bytes: Optional[bytes] = 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 +71,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 +87,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 +114,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 +144,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/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/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/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.15.dist-info → documente_shared-0.1.17.dist-info}/METADATA

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

{documente_shared-0.1.15.dist-info → documente_shared-0.1.17.dist-info}/RECORD

@@ -3,14 +3,14 @@ documente_shared/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 documente_shared/application/digest.py,sha256=Um6E8WfFri2_lly4RFWydJyvSfPZGFcOX-opEOzDCWc,172
 documente_shared/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/base_enum.py,sha256=DojAfn-zQdtjtImeHUpBzE6TBTm07XrbMOdW3h8RVd8,1449
-documente_shared/domain/entities.py,sha256=1TFf8OcZwNmwqCFwIfI-U4QzJHpeDJ-z88gKzhGQR0s,3997
-documente_shared/domain/enums.py,sha256=o20YJClja9FQbbZhdfreGCWYwjd3AH7q2kIzV9wn2dw,251
-documente_shared/domain/repositories.py,sha256=h_nArptP5Xh7Jrhden-Ii9rZQqzgLFHhuzj1p78LgmU,365
+documente_shared/domain/entities.py,sha256=YPYVVwn-vq8A1xiRC9kfERk3CE36cG7HTkRsbWri3TM,5188
+documente_shared/domain/enums.py,sha256=wrMpM-1KgXFkd-He-3onNcqR3Zf53Q-_axjXGjx7MIY,435
+documente_shared/domain/repositories.py,sha256=OJ6HKYx93RAHtj5M2uMk1P8cPdw7ubfhs1GMeUT_blU,540
 documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/infrastructure/dynamo_repositories.py,sha256=9t_ufQBSF7cJbeHYzfJABOQydj1HAvhI8UnqDEDPryA,532
+documente_shared/infrastructure/dynamo_repositories.py,sha256=0bnDxw-fQglpXXg5E4p0wyIbLn4gUy6nbusosveXpyU,777
 documente_shared/infrastructure/dynamo_table.py,sha256=5yF5EVQkHTZe4tLYZsfEDS8tp_iorMlMjNu-qn3Oj9E,2095
 documente_shared/infrastructure/s3_bucket.py,sha256=s4VWcl0HK6uhjjZmKefOYW5A3Dx74R9KCuMmXQKUXnQ,1582
 documente_shared/infrastructure/sqs_queue.py,sha256=PSiTAnjXvQ-W-9mzLpH2UjbQJTvYkMiaxNaMecF-cR4,1505
-documente_shared-0.1.15.dist-info/METADATA,sha256=3f5CIKiTUYnbleLbGnLWxeoYfKunoRcKeXcwBWbKCig,640
-documente_shared-0.1.15.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.15.dist-info/RECORD,,
+documente_shared-0.1.17.dist-info/METADATA,sha256=keoPKGD0e_OrqM4avuizzPx_jZdxhfnXJ0lOucLSB6c,640
+documente_shared-0.1.17.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.17.dist-info/RECORD,,