my-aws-helpers 4.1.0__tar.gz → 4.3.0__tar.gz

{my_aws_helpers-4.1.0 → my_aws_helpers-4.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my_aws_helpers
-Version: 4.1.0
+Version: 4.3.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-4.1.0 → my_aws_helpers-4.3.0}/my_aws_helpers/s3.py

@@ -2,6 +2,7 @@ import boto3
 import io
 import json
 from typing import Optional, Any, Dict
+from datetime import datetime, date
 from copy import copy
 import os
 import gzip
@@ -17,12 +18,29 @@ class ContentType(str, Enum):
     xml_content = "text/xml"
     json_content = "application/json"
     pdf_content = "application/pdf"
+    jpeg_content = "image/jpeg"
 
 
 class ContentEncoding(str, Enum):
     gzip = "gzip"
 
 
+class S3Serialiser:
+
+    @staticmethod
+    def _serialise(obj: Any):
+        if isinstance(obj, datetime) or isinstance(obj, date):
+            return obj.isoformat()
+        return obj
+
+    @staticmethod
+    def object_serialiser(obj: Any):
+        if isinstance(obj, list):
+            return [S3Serialiser.object_serialiser(obj=obj) for obj in obj]
+        if isinstance(obj, dict):
+            return {k: S3Serialiser.object_serialiser(v) for k, v in obj.items()}
+        return S3Serialiser._serialise(obj=obj)
+
 class S3Location:
     bucket: str
     file_name: str
@@ -186,7 +204,7 @@ class S3:
             logger.exception(f"Failed to save xml text to s3 due to {e}")
             return None
 
-    def save_xml_to_s3(self, pdf_content: bytes, s3_location: S3Location):
+    def save_pdf_to_s3(self, pdf_content: bytes, s3_location: S3Location):
         """
         pdf_content tends to come from:
             PyMuPdf.Document().write()
@@ -218,5 +236,18 @@ class S3:
             logger.exception(f"Failed to save dict to s3 due to {e}")
             return None
 
+    def save_jpeg_to_s3(self, content: bytes, s3_location: S3Location):
+        try:
+            return self.save_document_content(
+                file_contents=content,
+                s3_location=s3_location,
+                content_type=ContentType.jpeg_content.value,
+                compress=True,
+                content_encoding=ContentEncoding.gzip.value,
+            )
+        except Exception as e:
+            logger.exception(f"Failed to save jpeg to s3 due to {e}")
+            return None        
+
     def read_dict_from_s3(self, s3_location: S3Location) -> dict:
         return json.loads(self.read_binary_from_s3(s3_location=s3_location).decode("utf-8"))

{my_aws_helpers-4.1.0 → my_aws_helpers-4.3.0}/my_aws_helpers.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my-aws-helpers
-Version: 4.1.0
+Version: 4.3.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-4.1.0 → my_aws_helpers-4.3.0}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_namespace_packages, setup
 
 base_path = os.path.abspath(os.path.dirname(__file__))
 
-version = "4.1.0"
+version = "4.3.0"
 
 core = [
     "boto3==1.34.36",