documente_shared 0.1.54b0__py3-none-any.whl → 0.1.55__py3-none-any.whl

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
@@ -157,7 +160,6 @@ class DocumentProcessing(object):
             and self.completed_at == other.completed_at
         )
 
-
     @property
     def to_dict(self) -> dict:
         return {
@@ -172,6 +174,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 +218,7 @@ class DocumentProcessing(object):
             'file_bytes',
             'category',
             'sub_category',
+            'uploaded_from',
             'processed_csv_path',
             'processed_csv_bytes',
             'processed_xlsx_path',
@@ -251,6 +258,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/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.54b0.dist-info → documente_shared-0.1.55.dist-info}/METADATA

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

{documente_shared-0.1.54b0.dist-info → documente_shared-0.1.55.dist-info}/RECORD

@@ -8,15 +8,15 @@ 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=jOlMKFq12FgiYMJcQHku8IVwuOE5t-HEPuSV_zEeIFo,56
 documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/entities/document.py,sha256=X8uEw7TCy5o87-kscok2CcH67exsE6BrRqs6L-SvuTg,10111
+documente_shared/domain/entities/document.py,sha256=ZrtIzxS8SnMujYBCTURik4b0r1KsEPfGmBIH4QO0dog,10539
 documente_shared/domain/entities/document_metadata.py,sha256=Oa-c0xJODXaGtFkNxt9k86rSWGxY2LHVDOgu9tQxCGY,2354
-documente_shared/domain/enums.py,sha256=EA8RsuusX065Isp8qUTv9TPpWPkae6mbIC-yYENrkow,926
+documente_shared/domain/enums.py,sha256=zdy3kRWfKVxc-4wv5TLdQl-AbZkwsXEqlUZppCsMlsY,1797
 documente_shared/domain/repositories.py,sha256=g3qLUy2kT8esmvU4VxxSVnDaXeySKKQ7mUvIvxOwh9A,757
 documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 documente_shared/infrastructure/dynamo_repositories.py,sha256=SEad_HLppp2h_BKDSzb9oo1VlAVRZWelOPvJPlDwbzQ,1453
 documente_shared/infrastructure/dynamo_table.py,sha256=dK05KgFvIYCmOdMpq9-OV_OBrP6cCngiUikCJrxlwt4,2112
 documente_shared/infrastructure/s3_bucket.py,sha256=vT_yN42RFQXubtUn8ln-j13Os_-25UGClVtXg5Bkv6I,1932
 documente_shared/infrastructure/sqs_queue.py,sha256=PSiTAnjXvQ-W-9mzLpH2UjbQJTvYkMiaxNaMecF-cR4,1505
-documente_shared-0.1.54b0.dist-info/METADATA,sha256=X26yn3WiedJpJOzIwzdkIw9IIBg162aLmI_hsOFX2JM,802
-documente_shared-0.1.54b0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.54b0.dist-info/RECORD,,
+documente_shared-0.1.55.dist-info/METADATA,sha256=VTVTbWymdNgEYUd9Ylhnxrz1oecxpNVC9AWJ6d3l4QA,800
+documente_shared-0.1.55.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.55.dist-info/RECORD,,