documente_shared 0.1.3__tar.gz → 0.1.5__tar.gz

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

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

{documente_shared-0.1.3 → documente_shared-0.1.5}/documente_shared/infrastructure/dynamo_repositories.py

@@ -12,6 +12,5 @@ class DynamoDocumentProcessRepository(
         self.put(instance.to_dict)
         return instance
 
-
     def remove(self, instance: DocumentProcess):
-        self.delete(key=instance.digest)
+        self.delete(key={'digest': instance.digest})

{documente_shared-0.1.3 → documente_shared-0.1.5}/documente_shared/infrastructure/dynamo_table.py

@@ -44,7 +44,7 @@ class DynamoDBTable(object):
             ReturnValues=RETURN_VALUES,
         )
 
-    def delete(self, key: str):
+    def delete(self, key: dict):
         return self._table.delete_item(Key=key)
 
     def count(self) -> int:

{documente_shared-0.1.3 → documente_shared-0.1.5}/documente_shared/infrastructure/s3_bucket.py

@@ -1,7 +1,8 @@
+import boto3
+
 from dataclasses import dataclass
 from typing import Optional
 
-from python_layer.python import boto3
 
 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
@@ -9,28 +10,40 @@ def remove_none_values(data: dict) -> dict:  # noqa: WPS110
 
 @dataclass
 class S3Bucket(object):
-    bucket_url: str
+    bucket_name: str
 
     def __post_init__(self):
         self._resource = boto3.resource('s3')
 
-
-    def get(self, file_name: str):
-        return self._resource.Object(self.bucket_url, file_name).get()
+    def get(self, file_name: str) -> Optional[dict]:
+        try:
+            return self._resource.Object(self.bucket_name, file_name).get()
+        except self._resource.meta.client.exceptions.NoSuchKey:
+            return None
+
+    def get_bytes(self, file_name: str) -> Optional[dict]:
+        file_context = self.get(file_name)
+        if not file_context:
+            return None
+        return (
+            file_context['Body'].read()
+            if 'Body' in file_context
+            else None
+        )
 
     def upload(self, file_name, file_content, content_type: Optional[str] = None):
         optional_params = {'ContentType': content_type}
-        return self._resource.Object(self.bucket_url, file_name).put(
+        return self._resource.Object(self.bucket_name, file_name).put(
             Body=file_content,
             **remove_none_values(optional_params),
         )
 
     def delete(self, file_name):
-        return self._resource.Object(self.bucket_url, file_name).delete()
+        return self._resource.Object(self.bucket_name, file_name).delete()
 
     def get_url(self, file_name):
         return 'https://{bucket_url}.s3.amazonaws.com/{file_name}'.format(
-            bucket_url=self.bucket_url,
+            bucket_url=self.bucket_name,
             file_name=file_name,
         )
 

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.3 → documente_shared-0.1.5}/pyproject.toml

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