documente_shared 0.1.4__tar.gz → 0.1.5__tar.gz

{documente_shared-0.1.4 → documente_shared-0.1.5}/PKG-INFO

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

documente_shared-0.1.5/documente_shared/infrastructure/sqs_queue.py

@@ -0,0 +1,31 @@
+import json
+from dataclasses import dataclass
+
+import boto3
+
+
+@dataclass
+class SQSQueue(object):
+    queue_url: str
+
+    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}  # noqa: E501, WPS110
+        return self._client.send_message(**clean_params)

{documente_shared-0.1.4 → documente_shared-0.1.5}/pyproject.toml

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