pyaws-s3 1.0.21__tar.gz → 1.0.23__tar.gz

{pyaws_s3-1.0.21 → pyaws_s3-1.0.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyaws-s3
-Version: 1.0.21
+Version: 1.0.23
 Summary: A Python package for AWS S3 utilities
 Author-email: Giuseppe Zileni <giuseppe.zileni@gmail.com>
 Keywords: aws,s3,utilities

{pyaws_s3-1.0.21 → pyaws_s3-1.0.23}/pyaws_s3/s3.py

@@ -172,6 +172,65 @@ class S3Client:
         logger.info(f"Pre-signed URL: {temp_url}")
 
         return temp_url
+    
+    def upload_bytes(self, *args, **kwargs: Any):
+        """
+        Upload a Plotly Figure as a PNG image to an S3 bucket and generate a pre-signed URL.
+
+        Args:
+            fig (Figure): The Plotly Figure object to upload.
+            bucket_name (str): The name of the S3 bucket.
+            object_name (str): The name of the S3 object.
+
+        Keyword Args:
+            format_file (str): Format of the image. Defaults to 'png'.
+
+        Returns:
+            str: Pre-signed URL for the uploaded image.
+
+        Raises:
+            Exception: If there is an error uploading the image.
+        """
+        try:
+            
+            if args:
+                bytes_data = args[0] if len(args) > 0 else None
+                object_name = args[1] if len(args) > 1 else None
+            else:
+                bytes_data = kwargs.get("bytes_data", None) 
+                object_name = kwargs.get("object_name", None)
+            
+            if bytes_data is None:
+                raise Exception("Figure is None")
+            
+            if object_name is None:
+                raise Exception("Object name is None")
+
+            format_file : FormatFile = kwargs.get("format_file", "pdf")
+            mimetypes = "application/pdf"
+            
+            if format_file not in ["png", "jpeg", "svg", "html", "pdf"]:
+                raise Exception("Invalid format_file provided. Supported formats are: png, jpeg, svg, html, pdf")
+            if format_file == "png":
+                mimetypes = "image/png"
+            elif format_file == "jpeg":
+                mimetypes = "image/jpeg"
+            elif format_file == "svg":
+                mimetypes = "image/svg+xml"
+            elif format_file == "html":
+                mimetypes = "text/html"
+            elif format_file == "pdf":
+                mimetypes = "application/pdf"
+            else:
+                raise Exception("Invalid MIME type provided")
+            
+            s3_resource = self._get_s3_resource()
+            
+            s3_resource.Bucket(self.bucket_name).Object(object_name).put(Body=bytes_data, ContentType=mimetypes)
+
+        except Exception as e:
+            logger.error(f"Error uploading image: {str(e)}")
+            raise Exception(f"Error uploading image: {str(e)}")
 
     def upload_image(self, *args, **kwargs: Any) -> str:
         """

{pyaws_s3-1.0.21 → pyaws_s3-1.0.23}/pyaws_s3.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyaws-s3
-Version: 1.0.21
+Version: 1.0.23
 Summary: A Python package for AWS S3 utilities
 Author-email: Giuseppe Zileni <giuseppe.zileni@gmail.com>
 Keywords: aws,s3,utilities

{pyaws_s3-1.0.21 → pyaws_s3-1.0.23}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyaws-s3"
-version = "1.0.21"
+version = "1.0.23"
 description = "A Python package for AWS S3 utilities"
 authors = [
     { name="Giuseppe Zileni", email="giuseppe.zileni@gmail.com" }