documente_shared 0.1.131__py3-none-any.whl → 0.1.131b2__py3-none-any.whl

documente_shared/domain/entities/document.py

@@ -49,6 +49,7 @@ class DocumentProcessing(object):
     feedback: Optional[list | dict] = None
     completed_at: Optional[datetime] = None
     metadata: Optional[dict] = None
+    document_size: Optional[Decimal] = None
     metadata_items: Optional[List[DocumentProcessingMetadata]] = None
 
     def __post_init__(self):
@@ -108,7 +109,6 @@ class DocumentProcessing(object):
             DocumentProcessingStatus.COMPLETED,
             DocumentProcessingStatus.FAILED,
         ]
-
     @property
     def file_key(self) -> str:
         return remove_slash_from_path(self.file_path)
@@ -220,6 +220,12 @@ class DocumentProcessing(object):
     
     def in_review(self):
         self.status = DocumentProcessingStatus.IN_REVIEW
+    
+    def is_size_gt(self, size: Decimal = Decimal(10)):
+        return self.document_size and self.document_size >= size
+    
+    def is_size_lt(self, size: Decimal = Decimal(10)):
+        return self.document_size and self.document_size <= size
 
     def __eq__(self, other: 'DocumentProcessing') -> bool:
         if not other:
@@ -279,6 +285,7 @@ class DocumentProcessing(object):
             'metadata': self.metadata,
             'completed_at': self.completed_at.isoformat() if self.completed_at else None,
             'metadata_items': [metadata.to_dict for metadata in self.metadata_items],
+            'document_size': self.document_size,
         }
 
     @property
@@ -324,6 +331,7 @@ class DocumentProcessing(object):
             'failed_reason',
             'feedback',
             'metadata',
+            'document_size',
             'completed_at',
         ]
         for _property in instance_properties:
@@ -377,5 +385,6 @@ class DocumentProcessing(object):
                 DocumentProcessingMetadata.from_dict(metadata)
                 for metadata in data.get('metadata_items', [])
             ],
+            document_size=data.get('document_size'),
         )
 

documente_shared/domain/entities/scaling.py

@@ -4,14 +4,16 @@ from typing import Optional
 
 @dataclass
 class ScalingRequirements(object):
-    documents: Optional[int] = 0
+    lg_documents: Optional[int] = 0
+    md_documents: Optional[int] = 0
     processing_cases: Optional[int] = 0
     processing_case_items: Optional[int] = 0
 
     @property
     def to_dict(self):
         return {
-            "documents": self.documents,
+            "lg_documents": self.lg_documents,
+            "md_documents": self.md_documents,
             "processing_cases": self.processing_cases,
             "processing_case_items": self.processing_case_items,
         }
@@ -19,7 +21,8 @@ class ScalingRequirements(object):
     @classmethod
     def from_dict(cls, data: dict) -> "ScalingRequirements":
         return cls(
-            documents=data.get("documents", 0),
+            lg_documents=data.get("lg_documents", 0),
+            md_documents=data.get("md_documents", 0),
             processing_cases=data.get("processing_cases", 0),
             processing_case_items=data.get("processing_case_items", 0),
         )

{documente_shared-0.1.131.dist-info → documente_shared-0.1.131b2.dist-info}/METADATA

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

{documente_shared-0.1.131.dist-info → documente_shared-0.1.131b2.dist-info}/RECORD

@@ -14,7 +14,7 @@ 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=NG5BGaXBr_FnzudjTRPxpDpyiSDdaB_PLCdlYlFUQeU,187
 documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/entities/document.py,sha256=eEEoAujD_wjFdcSvwB1WW6QbCNQSpijc0UfaMlytprk,13464
+documente_shared/domain/entities/document.py,sha256=Os901LfH8FPuFxOHj_RtTmqysi3huUYyvdeM-4VKNeA,13888
 documente_shared/domain/entities/document_metadata.py,sha256=ygyFIC5qwxlm8DUM5kvVFny9zJfPQS8vNLM2br5XsQ8,2353
 documente_shared/domain/entities/in_memory_document.py,sha256=6LlUtF9YbFPNxP-ytrWkFYQzvdJJJ9iC8X4_U125hEE,2113
 documente_shared/domain/entities/processing_case.py,sha256=D57zP704mBtt563Pbmd41PFQSna4chVWsWTR_2ZAtTU,6602
@@ -23,7 +23,7 @@ documente_shared/domain/entities/processing_case_item.py,sha256=IN46jCGIv_DXEp1H
 documente_shared/domain/entities/processing_case_item_filters.py,sha256=R_AvDCB496Lww1qn2OwtltqULKE3IpcJB0ejnmRkg7Q,2009
 documente_shared/domain/entities/processing_documents.py,sha256=YYuTkdCNkqlO8cA0onJsZYtstxGt9M5NMuIO_87lB14,352
 documente_shared/domain/entities/processing_event.py,sha256=sejo9eM4A6fDDFq3qCX3V1GMGwQVUh_5oJaPDiUwpoc,2350
-documente_shared/domain/entities/scaling.py,sha256=Me1z3X-5NjzPMX-TBQ4xHwE_44tIJegi1QSCHQRNtx4,745
+documente_shared/domain/entities/scaling.py,sha256=DPOE_W4NZ903cyNckmDPus_sYWEZfnvn5ar3Zf5JM3Q,897
 documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/domain/enums/common.py,sha256=tgM098gKYHCp8LVNA6FKd-O_Meoi0Cx_UjQ3OKLxGYY,3127
 documente_shared/domain/enums/document.py,sha256=QwvckW-VJBSujllIVloKlZUh1pI5UnX4oueYbV5CYGw,3205
@@ -55,6 +55,6 @@ documente_shared/infrastructure/services/http_scaling.py,sha256=cIo-61nfIwbtO86E
 documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
 documente_shared/presentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/presentation/presenters.py,sha256=GGAEwefmjCIVepsUA2oZOVLxXbhhiISPM0Jgt6dT6O0,423
-documente_shared-0.1.131.dist-info/METADATA,sha256=AXX_tT5o0nB_QV8wrFHu8SE5lk0mKvZMKyFF8M0_Xqw,963
-documente_shared-0.1.131.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.131.dist-info/RECORD,,
+documente_shared-0.1.131b2.dist-info/METADATA,sha256=d3cksEy0mID1ZIAE7VrOVINYuKPoMjNnHvSqfHjOvpU,965
+documente_shared-0.1.131b2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.131b2.dist-info/RECORD,,