documente_shared 0.1.54b1__tar.gz → 0.1.56__tar.gz

{documente_shared-0.1.54b1 → documente_shared-0.1.56}/PKG-INFO

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

{documente_shared-0.1.54b1 → documente_shared-0.1.56}/documente_shared/domain/entities/document.py

@@ -9,14 +9,16 @@ from documente_shared.domain.entities.document_metadata import DocumentProcessin
 from documente_shared.domain.enums import (
     DocumentProcessingStatus,
     DocumentProcessingSubCategory,
-    DocumentProcessingCategory,
+    DocumentProcessingCategory, DocumentProcessingSource,
 )
 
+
 def remove_slash_from_path(path: str) -> str:
     if path and path.startswith('/'):
         return path[1:]
     return path
 
+
 @dataclass
 class DocumentProcessing(object):
     digest: str
@@ -25,6 +27,7 @@ class DocumentProcessing(object):
     file_bytes: Optional[bytes] = None
     category: Optional[DocumentProcessingCategory] = None
     sub_category: Optional[DocumentProcessingSubCategory] = None
+    uploaded_from: Optional[DocumentProcessingSource] = None
     processed_csv_path: Optional[str] = None
     processed_csv_bytes: Optional[bytes] = None
     processed_xlsx_path: Optional[str] = None
@@ -85,6 +88,10 @@ class DocumentProcessing(object):
             DocumentProcessingStatus.FAILED,
         ]
 
+    def pending(self):
+        self.status = DocumentProcessingStatus.PENDING
+        self.started_at = None
+
     def enqueue(self, timezone: tzinfo = la_paz_tz):
         self.status = DocumentProcessingStatus.ENQUEUED
         self.enqueued_at = datetime.now(tz=timezone)
@@ -157,7 +164,6 @@ class DocumentProcessing(object):
             and self.completed_at == other.completed_at
         )
 
-
     @property
     def to_dict(self) -> dict:
         return {
@@ -172,6 +178,10 @@ class DocumentProcessing(object):
                 str(self.sub_category)
                 if self.sub_category else None
             ),
+            'uploaded_from': (
+                str(self.uploaded_from)
+                if self.uploaded_from else None
+            ),
             'processed_csv_path': self.processed_csv_path,
             'processed_xlsx_path': self.processed_xlsx_path,
             'processed_json_path': self.processed_json_path,
@@ -212,6 +222,7 @@ class DocumentProcessing(object):
             'file_bytes',
             'category',
             'sub_category',
+            'uploaded_from',
             'processed_csv_path',
             'processed_csv_bytes',
             'processed_xlsx_path',
@@ -251,6 +262,10 @@ class DocumentProcessing(object):
                 DocumentProcessingSubCategory.from_value(data.get('sub_category'))
                 if data.get('sub_category') else None
             ),
+            uploaded_from=(
+                DocumentProcessingSource.from_value(data.get('uploaded_from'))
+                if data.get('uploaded_from') else None
+            ),
             processed_csv_path=data.get('processed_csv_path'),
             processed_xlsx_path=data.get('processed_xlsx_path'),
             processed_json_path=data.get('processed_json_path'),

{documente_shared-0.1.54b1 → documente_shared-0.1.56}/documente_shared/domain/enums.py

@@ -11,6 +11,7 @@ class DocumentProcessingStatus(BaseEnum):
     CANCELLED = 'CANCELLED'
     IN_REVIEW = 'IN_REVIEW'
 
+
 class DocumentProcessingCategory(BaseEnum):
     CIRCULAR = 'CIRCULAR'
     DESGRAVAMEN = 'DESGRAVAMEN'
@@ -24,6 +25,39 @@ class DocumentProcessingCategory(BaseEnum):
         return self == DocumentProcessingCategory.DESGRAVAMEN
 
 
+class DocumentProcessingSource(BaseEnum):
+    AGENT_UI = 'AGENT_UI'
+    AGENT_CRAWLER = 'AGENT_CRAWLER'
+    PLATFORM_UI = 'PLATFORM_UI'
+    PLATFORM_API = 'PLATFORM_API'
+    AWS_CONSOLE = 'AWS_CONSOLE'
+    LOCAL_MANUAL = 'LOCAL_MANUAL'
+
+    @property
+    def is_agent_ui(self):
+        return self == DocumentProcessingSource.AGENT_UI
+
+    @property
+    def is_agent_crawler(self):
+        return self == DocumentProcessingSource.AGENT_CRAWLER
+
+    @property
+    def is_platform_ui(self):
+        return self == DocumentProcessingSource.PLATFORM_UI
+
+    @property
+    def is_platform_api(self):
+        return self == DocumentProcessingSource.PLATFORM_API
+
+    @property
+    def is_aws_console(self):
+        return self == DocumentProcessingSource.AWS_CONSOLE
+
+    @property
+    def is_local_manual(self):
+        return self == DocumentProcessingSource.LOCAL_MANUAL
+
+
 class DocumentProcessingSubCategory(BaseEnum):
     # Circulares
     CC_COMBINADA = 'CC_COMBINADA'

{documente_shared-0.1.54b1 → documente_shared-0.1.56}/pyproject.toml

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