documente_shared 0.1.72b0__py3-none-any.whl → 0.1.74__py3-none-any.whl

documente_shared/infrastructure/s3_bucket.py

@@ -1,57 +1,57 @@
-import boto3
-
-from dataclasses import dataclass
-from typing import Optional
-
-from documente_shared.domain.entities.document import remove_slash_from_path
-
-
-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')
-
-    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'].read()
-            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()
+import boto3
+
+from dataclasses import dataclass
+from typing import Optional
+
+from documente_shared.domain.entities.document import remove_slash_from_path
+
+
+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')
+
+    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'].read()
+            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/sqs_queue.py

@@ -1,48 +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,
-        )
+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-0.1.72b0.dist-info → documente_shared-0.1.74.dist-info}/METADATA

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

documente_shared-0.1.74.dist-info/RECORD

@@ -0,0 +1,42 @@
+documente_shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+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=ADiWi6Mk3YQGx3boGsDqdb5wk8qmabkGRy7bhNFa1OY,649
+documente_shared/application/query_params.py,sha256=JscPqFBx28p-x9i2g6waY7Yl4FQM1zn2zSbEoTrkK1k,3938
+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=AthTUyA-QZE3WAT7lMoKVr_Z8mO_3qERuCnZge0DTLQ,12595
+documente_shared/domain/entities/document_metadata.py,sha256=ygyFIC5qwxlm8DUM5kvVFny9zJfPQS8vNLM2br5XsQ8,2353
+documente_shared/domain/entities/in_memory_result.py,sha256=Q1E9vnLL5Hz5xunOqWtQmJOMjoK5KN42LZr18GlBAZo,1246
+documente_shared/domain/entities/processing_case.py,sha256=3bL6BgFwWe6F5keZ6A1K_lLsrwGpkcKg98roM_i6kEQ,5167
+documente_shared/domain/entities/processing_case_filters.py,sha256=146fnvzB-GvDPEzFHX3ckaFM9nLqG-4deUaEhl7Epwg,1784
+documente_shared/domain/entities/processing_case_item.py,sha256=hglyPhZBSjlMKVdEVccyHXU3aQRdZ8hfZlXDJnkf7wA,7996
+documente_shared/domain/entities/processing_case_item_filters.py,sha256=YUsMk5BTtjUNzKy-nlkZg4NLPlPT1MnhP49rLhwlyB8,1891
+documente_shared/domain/entities/processing_event.py,sha256=m1O0gcNaE_SszeIhxM3uYPHSpyOUmize6mfRw1_bYZo,1723
+documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/domain/enums/common.py,sha256=vXldMUPhhWo0PfTgYwDSjI8bur_lYcImZYiV7yAO7DQ,2262
+documente_shared/domain/enums/document.py,sha256=NltZA1YVgJ7dVfSQdJFIE0ZUGf9Y-nxNXsVQ6GiPLL4,1827
+documente_shared/domain/enums/processing_case.py,sha256=LhFhcoWlockxcpplsVdC6M2kpXn7sOdzQySf24wFhx8,1572
+documente_shared/domain/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/domain/repositories/document.py,sha256=vJzr6c92gqBzyhaEdjrvnoneKRrWmJ0AsvocPnhxiLU,767
+documente_shared/domain/repositories/processing_case.py,sha256=6gvQPHUORGuBL8Mr5OPYYOmo1oK2IjAdvBrbd0yvBbs,749
+documente_shared/domain/repositories/processing_case_item.py,sha256=8yLF4cQizqb6CuoUNVe-a7ldzZC2xaXPTQ_M-VdlUJM,867
+documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/infrastructure/documente_client.py,sha256=UjVWs9DKa-yhw5DVbcEs8iJxalHOarusVayi_ob6QhE,494
+documente_shared/infrastructure/dynamo_table.py,sha256=TMQbcuty7wjDMbuhI8PbT0IGXelgELsNTtqTEQeZ824,2112
+documente_shared/infrastructure/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+documente_shared/infrastructure/repositories/dynamo_document.py,sha256=_Yp4gtA-n-hJ2w2wAM5BMCs2Mf46Q2Kq3eHqlxudkL4,1443
+documente_shared/infrastructure/repositories/dynamo_processing_case.py,sha256=UJcROnmfKvE5k_GZOHvaYDvnNSX4Rm22xCB7tUaUUEU,1386
+documente_shared/infrastructure/repositories/dynamo_processing_case_item.py,sha256=bqD6uFoMmG_CgLcY4Q5VLGVD5A1bA927_dzllHTZ7O8,1447
+documente_shared/infrastructure/repositories/http_processing_case.py,sha256=QUG3_mD3nsi0mqHqt-udM9CCYNY6DWJ0E6_Fx8cLt54,1772
+documente_shared/infrastructure/repositories/http_processing_case_item.py,sha256=vD_L6hBRv2qz3UnxqmBLjTQARY21xPDmsvM4oAUUf0s,1780
+documente_shared/infrastructure/s3_bucket.py,sha256=vT_yN42RFQXubtUn8ln-j13Os_-25UGClVtXg5Bkv6I,1932
+documente_shared/infrastructure/sqs_queue.py,sha256=KZWeHZ9zmXmrxoNpOQX7GEdDhZ1knbPXgwSwFwJblGg,1504
+documente_shared-0.1.74.dist-info/METADATA,sha256=CZ6pj4HbFF6x2Nu-qHiUEcMbfxNjaLg3GBIr3sWoYnI,881
+documente_shared-0.1.74.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+documente_shared-0.1.74.dist-info/RECORD,,

documente_shared/infrastructure/repositories/http_document_processing.py

@@ -1,41 +0,0 @@
-from dataclasses import dataclass
-from typing import List, Optional
-
-from documente_shared.domain.entities.documet import DocumentProcessing
-from documente_shared.domain.enums.common import ProcessingStatus, DocumentProcessingCategory
-from documente_shared.domain.repositories.document_processing import DocumentProcessingRepository
-from documente_shared.infrastructure.documente_client import DocumenteClientMixin
-
-class HttpDocumentProcessingRepository(
-    DocumenteClientMixin,
-    DocumentProcessingRepository,
-):
-    def find(self, digest: str) -> Optional[DocumentProcessing]:
-        response = self.session.get(f"{self.base_url}/documents/{digest}/")
-        if response.status_code == 200:
-            return DocumentProcessing.from_dict(response.json())
-        return None
-
-    def persist(self, instance: DocumentProcessing) -> DocumentProcessing:
-        response = self.session.put(
-            url=f"{self.base_url}/documents/{instance.uuid}/",
-            json=instance.to_simple_dict,
-        )
-        if response.status_code in [200, 201]:
-            return DocumentProcessing.from_dict(response.json())
-        return instance
-    
-    def remove(self, instance: DocumentProcessing):
-        self.session.delete(f"{self.base_url}/documents/{instance.uuid}/")
-        
-    def filter(self, statuses: List[ProcessingStatus]) -> List[DocumentProcessing]:
-        response = self.session.get(f"{self.base_url}/processing-cases/{statuses}/")
-        if response.status_code == 200:
-            raw_response = response.json()
-            return [
-                DocumentProcessing.from_dict(item)
-                for item in raw_response.get('data', [])
-            ]
-        return []
-    
-    

documente_shared-0.1.72b0.dist-info/RECORD

@@ -1,40 +0,0 @@
-documente_shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/application/digest.py,sha256=9T5e5SC1EKYrIjcmzqnlG29pyjJl29q0kdNl-zgsmws,179
-documente_shared/application/exceptions.py,sha256=zrV-wPYXLFUQhA_0hoXKnFAhNntNG17oGB_MU_YhtIg,677
-documente_shared/application/files.py,sha256=EOm6TQWXXjlX-6dQeY2__GdVrzeMDZVnoPU_bVtZDB0,671
-documente_shared/application/time_utils.py,sha256=YBgiJfVdLo2IlgS2Kxp24DY5AwSByDqZ3QjiaOwPKGs,448
-documente_shared/application/timezone.py,sha256=RlLmdvGlMJ-TSfGrTRm1wlJPwpc-_bu9RtWwdKYIPfo,190
-documente_shared/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/base_enum.py,sha256=KihiZGq1ylvQaWjEWtPHCAw9uFpFWsu8eJtLB_nwaWs,1502
-documente_shared/domain/constants.py,sha256=lws6A8b4KM2ch9DAh4Xvpvi_1oWNf_cc0kL3NwpMUOc,58
-documente_shared/domain/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/entities/document.py,sha256=8iLXzJ0qcy1LAgztsTyOzk_DthgJddH5xyTATS5sShw,12943
-documente_shared/domain/entities/document_metadata.py,sha256=QXFy0UemTrECVUFWpsE9o-ZI5Va7QHjvyhNA55a-n9s,2416
-documente_shared/domain/entities/in_memory_result.py,sha256=wG_yWdDglt3BMW0mtc2YFZwBeTrx4_2H92IuqaMK01Q,1297
-documente_shared/domain/entities/processing_case.py,sha256=RlvG2ioQ439PeKcpjczLcwipMYsOVOriT6B1aPzW0OM,5301
-documente_shared/domain/entities/processing_case_item.py,sha256=4TIAJPviUSY77pHWlXZ36R3hkziuHiKpZj9YsXA5NEs,8212
-documente_shared/domain/entities/processing_event.py,sha256=oCi5ym5HcmACKFYoExXEq8pgVAskx7jAttpsvptTGLQ,1772
-documente_shared/domain/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/enums/common.py,sha256=fxLkGAfJStNmJDBN6jyLXDD_d3T65DZqZCgSFVRDQ0s,2357
-documente_shared/domain/enums/document.py,sha256=GlAUn_Rq0wzf98iQQNBxQRO6n-0_1LjRm6zYIi2irj0,1898
-documente_shared/domain/enums/processing_case.py,sha256=3cJ8gLcqWqKtohW1kd1PiGGXx13SkY80mnSTInDutMA,1600
-documente_shared/domain/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/domain/repositories/document.py,sha256=SSqc3zuVpCfnr3OdRj5huREiOYvlID7JSmuFHDtEeg4,791
-documente_shared/domain/repositories/processing_case.py,sha256=xF16EoarAaSmH54w50yBaDwNWrAasZXCEb2q2gxKoOk,753
-documente_shared/domain/repositories/processing_case_item.py,sha256=zOZvjc4Vfc9avgMNr0GEMiCD5fgKB9KJUVhRalbA_Rs,986
-documente_shared/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/infrastructure/documente_client.py,sha256=iuwpNBa9H02a36xqDRt8ffT3KlIMkHkV4mt0RtuvZXo,515
-documente_shared/infrastructure/dynamo_table.py,sha256=Kdz2iViRb5C7wlfbBDp-JDcu2dq_MY1io3gB1uLff0A,2187
-documente_shared/infrastructure/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-documente_shared/infrastructure/repositories/dynamo_document.py,sha256=-zf33Dey17_IPsqpUeczZyD2MGzzK9LfV6j-8iQLYPY,1486
-documente_shared/infrastructure/repositories/dynamo_processing_case.py,sha256=kV1D6T19BW9veUyDIgcijelndiGO5e1tOinBGV8S908,1444
-documente_shared/infrastructure/repositories/dynamo_processing_case_item.py,sha256=AghVYUMr9sFG3-pz1A2YctBDxKe1P9447tjhomdHbWI,1865
-documente_shared/infrastructure/repositories/http_document_processing.py,sha256=vDFFz3yCLscXt-lyMSNOFMeZdfq6D4lLj_ndG4CSNGM,1752
-documente_shared/infrastructure/repositories/http_processing_case.py,sha256=oRDn-A2HCDJsEfDL4lyh4jrTQ_oTLEazCBQeIUYnYKw,1675
-documente_shared/infrastructure/repositories/http_processing_case_item.py,sha256=LZXf_FG9OZT51DbOYPupT1pGdhRsC8IOEo1nnib4yeo,2244
-documente_shared/infrastructure/s3_bucket.py,sha256=qPCBpheM1qvtC08vynlwIGJZCCOhefFcD3Q3ClusXmM,1989
-documente_shared/infrastructure/sqs_queue.py,sha256=UcjrT23JpK1Em9f0qbxI7oYnCPUD7Ll1aJ4JICcqPRo,1551
-documente_shared-0.1.72b0.dist-info/METADATA,sha256=0QOi_tenMNop-s7L3Grz-u3PnFUrxdUQwupWownkeGg,883
-documente_shared-0.1.72b0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-documente_shared-0.1.72b0.dist-info/RECORD,,