documente_shared 0.1.145__py3-none-any.whl

documente_shared/infrastructure/s3_bucket.py

@@ -0,0 +1,58 @@
+import boto3
+
+from dataclasses import dataclass
+from typing import Optional
+
+from documente_shared.domain.entities.document import remove_slash_from_path
+from documente_shared.application.retry_utils import retry_on_size_integrity
+
+def remove_none_values(data: dict) -> dict:  # noqa: WPS110
+    return {key: value for key, value in data.items() if value is not None}  # noqa: WPS110
+
+
+@dataclass
+class S3Bucket(object):
+    bucket_name: str
+
+    def __post_init__(self):
+        self._resource = boto3.resource('s3')
+
+    @retry_on_size_integrity()
+    def get(self, file_key: str) -> Optional[dict]:
+        try:
+            return self._resource.Object(self.bucket_name, file_key).get()
+        except self._resource.meta.client.exceptions.NoSuchKey:
+            return None
+
+    def get_bytes(self, file_key: str) -> Optional[bytes]:
+        cleaned_file_key = remove_slash_from_path(file_key)
+        file_context = self.get(cleaned_file_key)
+        if not file_context:
+            return None
+        return (
+            file_context['Body']
+            if 'Body' in file_context
+            else None
+        )
+
+    def upload(self, file_key: str, file_content, content_type: Optional[str] = None):
+        cleaned_file_key = remove_slash_from_path(file_key)
+        optional_params = {'ContentType': content_type}
+        return self._resource.Object(self.bucket_name, cleaned_file_key).put(
+            Body=file_content,
+            **remove_none_values(optional_params),
+        )
+
+    def delete(self, file_key: str):
+        cleaned_file_key = remove_slash_from_path(file_key)
+        return self._resource.Object(self.bucket_name, cleaned_file_key).delete()
+
+    def get_url(self, file_key: str):
+        cleaned_file_key = remove_slash_from_path(file_key)
+        return 'https://{bucket_url}.s3.amazonaws.com/{file_key}'.format(
+            bucket_url=self.bucket_name,
+            file_key=cleaned_file_key,
+        )
+
+    def read(self, file_key: str) -> bytes:
+        return self.get(file_key)['Body'].read()

documente_shared/infrastructure/services/http_scaling.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass
+
+from loguru import logger
+
+from documente_shared.application.payloads import camel_to_snake
+from documente_shared.domain.entities.scaling import ScalingRequirements
+from documente_shared.domain.interfaces.scaling import ScalingService
+from documente_shared.infrastructure.documente_client import DocumenteClientMixin
+
+
+@dataclass
+class HttpScalingService(
+    DocumenteClientMixin,
+    ScalingService,
+):
+    def get_requirements(self) -> ScalingRequirements:
+        response = self.session.get(f"{self.api_url}/v1/scaling-requirements/")
+        if response.status_code == 200:
+            response_json = response.json()
+            return ScalingRequirements.from_dict(
+                data=camel_to_snake(response_json.get('data', {}))
+            )
+
+        logger.warning(f'Error getting scaling requirements: {response.text}')
+        return ScalingRequirements()

documente_shared/infrastructure/sqs_queue.py

@@ -0,0 +1,48 @@
+import json
+import boto3
+
+from dataclasses import dataclass
+
+
+@dataclass
+class SQSQueue(object):
+    queue_url: str
+    visibility_timeout: int = 60 * 10
+    waiting_timeout: int = 20
+
+    def __post_init__(self):
+        self._client = boto3.client('sqs')
+
+    def send_message(
+        self,
+        payload: dict,
+        message_attributes: dict = None,
+        delay_seconds: dict = None,
+        message_group_id: dict = None,
+        message_deduplication_id: dict =None,
+    ):
+        message_params = {
+            'QueueUrl': self.queue_url,
+            'MessageBody': json.dumps(payload),
+            'MessageAttributes': message_attributes,
+            'DelaySeconds': delay_seconds,
+            'MessageGroupId': message_group_id,
+            'MessageDeduplicationId': message_deduplication_id,
+        }
+        clean_params = {key: value for key, value in message_params.items() if value}
+        return self._client.send_message(**clean_params)
+
+    def delete_message(self, receipt_handle: str):
+        return self._client.delete_message(
+            QueueUrl=self.queue_url,
+            ReceiptHandle=receipt_handle
+        )
+
+    def fetch_messages(self, num_messages: int = 1) -> list[dict]:
+        response = self._client.receive_message(
+            QueueUrl=self.queue_url,
+            MaxNumberOfMessages=num_messages,
+            VisibilityTimeout=self.visibility_timeout,
+            WaitTimeSeconds=self.waiting_timeout,
+        )
+        return response.get('Messages', [])

documente_shared/presentation/presenters.py

@@ -0,0 +1,16 @@
+from dataclasses import dataclass
+
+from documente_shared.application.dates import utc_now
+from documente_shared.domain.enums.common import TaskResultStatus
+
+
+@dataclass
+class TaskResultPresenter(object):
+    status: TaskResultStatus = TaskResultStatus.SUCCESS
+
+    @property
+    def to_dict(self) -> dict:
+        return {
+            "status": str(self.status),
+            "completed_at": utc_now().isoformat(),
+        }

documente_shared-0.1.145.dist-info/METADATA

@@ -0,0 +1,39 @@
+Metadata-Version: 2.1
+Name: documente_shared
+Version: 0.1.145
+Summary: Shared utilities for Documente AI projects
+License: MIT
+Author: Tech
+Author-email: tech@llamitai.com
+Requires-Python: >=3.10,<3.13
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: boto3 (>=1.37.19,<2.0.0)
+Requires-Dist: botocore (>=1.37.19,<2.0.0)
+Requires-Dist: loguru (>=0.7.3,<0.8.0)
+Requires-Dist: pytz (>=2025.2,<2026.0)
+Requires-Dist: requests (>=2.32.3,<3.0.0)
+Requires-Dist: sentry-sdk (>=2.19.2,<3.0.0)
+Requires-Dist: unidecode (>=1.3.8,<2.0.0)
+Description-Content-Type: text/markdown
+
+
+# Documente Shared
+
+Utilidades para proyectos Documente AI
+
+## Instalación
+
+```bash
+make build
+```
+
+## Publicación
+Publica `documente-shared` en PyPi
+
+```bash
+make publish
+```

documente_shared-0.1.145.dist-info/RECORD

@@ -0,0 +1,63 @@
+documente_shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/application/dates.py,sha256=uExNddWmX9VEX_u420JGoC7fL-ieJ4956I0aw_WX3sQ,109
+documente_shared/application/digest.py,sha256=Um6E8WfFri2_lly4RFWydJyvSfPZGFcOX-opEOzDCWc,172
+documente_shared/application/exceptions.py,sha256=lQM8m7wmI9OTLGva0gd7s7YT7ldaTk_Ln4t32PpzNf8,654
+documente_shared/application/files.py,sha256=_qkpeFPVmHyNhD6_PFZdzhyYvshJVeeSiwuTggK-A9Q,793
+documente_shared/application/json.py,sha256=5y67-DoiJlq_fLefgJ8YBVsdyua4KykxbSUKQqcwnnQ,1223
+documente_shared/application/numbers.py,sha256=rik1SqMxLzXNL2S5Yh1Q8CSONdcjjoCBnByFasxR4_Q,212
+documente_shared/application/payloads.py,sha256=gLVTGM1gPfcaRXQMddGRKK0qSazMvyribpRB7WADTNg,922
+documente_shared/application/query_params.py,sha256=dy-OcCzLt4ZDseMm5vB3uzYzFFjEiit8nKaTzODUGPg,4025
+documente_shared/application/retry_utils.py,sha256=8Cj8oQp8mUynEeQNOyH_FRwODTku8KyljijYVxW7_sw,3140
+documente_shared/application/time_utils.py,sha256=_fxgh8VoGPkdsft47COJ16vFwt8pMbHIJCgDFHLSlrU,435
+documente_shared/application/timezone.py,sha256=NHpzTzOPD_fWQiJ4BrRqt_TIDs5XyB5ZMR7x8vUk8gQ,183
+documente_shared/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+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=RSpKwP2x5KRqXezhJ25NpQs7-QpH_6EdBayARXYLi8w,14696
+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=acfkT5aadsfFFf4Ihwgm8xOc7PaGLH_326I8ppLpY54,7554
+documente_shared/domain/entities/processing_case_filters.py,sha256=harKyu7QEuL1bI_Z8_UxkVCMo5r9vHeNHyi_Ja07vjs,1953
+documente_shared/domain/entities/processing_case_item.py,sha256=IN46jCGIv_DXEp1HQbqZiDteVS2UzEob7WLS4VXft1s,10586
+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=j_-6RL9cvZZukA6B7I17OuwNfUfHUfuSTt2wMsEXNl8,1019
+documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/domain/enums/circular_oficio.py,sha256=iHwWaM8uEYmVhAFZU1xpAdDQPlIn_el2uL3WYX2I9X4,732
+documente_shared/domain/enums/common.py,sha256=tgM098gKYHCp8LVNA6FKd-O_Meoi0Cx_UjQ3OKLxGYY,3127
+documente_shared/domain/enums/document.py,sha256=QwvckW-VJBSujllIVloKlZUh1pI5UnX4oueYbV5CYGw,3205
+documente_shared/domain/enums/document_type_record.py,sha256=yAs-L1znL94IG2rjJ7Bcu-5vIPkr8eZP0ZN1fg2Ksbo,424
+documente_shared/domain/enums/processing_case.py,sha256=IF8ivf42TY67Vg9-3SYJvmptzsskMtlqPCUNfekoq10,1794
+documente_shared/domain/exceptions.py,sha256=S2xkkFPiR8jVmA6UhRb-bPVrm6JvohXw2MUhZOzdnRo,58
+documente_shared/domain/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/domain/interfaces/scaling.py,sha256=yWAud0rcLKMMf2-QxEE__GRpzqUY0H9_qxlQTfnRYmw,249
+documente_shared/domain/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/domain/repositories/document.py,sha256=EZCbz8ManC1C2JcGuQQIJKON28AUAnUITZjy2zotRbY,819
+documente_shared/domain/repositories/processing_case.py,sha256=1vGe4084q6AWaIEx8GgVAu0qm1gkResVyH5UZxmfNi4,1126
+documente_shared/domain/repositories/processing_case_item.py,sha256=iUWD_tZWifB7gEDjVZL1CIFAu59q7lWkcJ4Kic5cNvc,1516
+documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/infrastructure/documente_client.py,sha256=u6k73yIefuEkCsHULMpVCydFzrSXclf_5oQd2D7CkKU,698
+documente_shared/infrastructure/dynamo_table.py,sha256=TMQbcuty7wjDMbuhI8PbT0IGXelgELsNTtqTEQeZ824,2112
+documente_shared/infrastructure/lambdas.py,sha256=sGgkw7Mhvuq2TpbW_RNdf5JvQnuzxWYH6gPOVtQ4DtE,357
+documente_shared/infrastructure/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/infrastructure/repositories/dynamo_document.py,sha256=CU3bljm83TwHO4YSUkFDUI0RPhV1yCllpNSL8_8dwBo,1495
+documente_shared/infrastructure/repositories/dynamo_processing_case.py,sha256=bQQWjRknHAogXhabK1aCeYyiB73HWUX3t_TCpCx9ITo,1794
+documente_shared/infrastructure/repositories/dynamo_processing_case_item.py,sha256=gD9WCA3sizp74AaWSzk-eIeZJj2_et7yJ6Q0PNQnoWg,2297
+documente_shared/infrastructure/repositories/http_document.py,sha256=Qc5D5Q0fpHYLyxOMlBYMdnRhGM-lF7ZNqQxZyClsvsA,2666
+documente_shared/infrastructure/repositories/http_processing_case.py,sha256=qO5O5zADNVj1ibHGgoGEIEiGbGovgMIwmMzYQ0S7ALA,3074
+documente_shared/infrastructure/repositories/http_processing_case_item.py,sha256=Txmke8k2krMfuc3Sj9mabRw4OQDaVOtB5Gp9PPTOZ78,4458
+documente_shared/infrastructure/repositories/mem_document.py,sha256=L4ebQljMKixnM_qI0StszUUYCCTHA_WAXW3RbPdYJFs,1559
+documente_shared/infrastructure/repositories/mem_processing_case.py,sha256=ayyhtXGHLIIhtlTUjlr1s4AKn01QBFW6zKNYYL138J8,1464
+documente_shared/infrastructure/repositories/mem_processing_case_item.py,sha256=kGURB2dmnhpRWxzxLCdQmhzsMAboGEKLyI3_F9UV2AQ,1566
+documente_shared/infrastructure/s3_bucket.py,sha256=yaT8hkgNUk5Q6ts-E2Igo4b6nBldFi8KE_dFnJyt4JA,2032
+documente_shared/infrastructure/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/infrastructure/services/http_scaling.py,sha256=cIo-61nfIwbtO86EGi5r1tFi9g3VJXldhguddt4JUyc,906
+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.145.dist-info/METADATA,sha256=yzh_s94WhZRwCe3LDGkfkXnPxB9bjvUpbvEulRBwuMc,963
+documente_shared-0.1.145.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.145.dist-info/RECORD,,

documente_shared-0.1.145.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.9.1
+Root-Is-Purelib: true
+Tag: py3-none-any