boto3-assist 0.2.2__tar.gz → 0.2.3__tar.gz

{boto3_assist-0.2.2 → boto3_assist-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: boto3_assist
-Version: 0.2.2
+Version: 0.2.3
 Summary: Additional boto3 wrappers to make your life a little easier
 Author-email: Eric Wilson <boto3-assist@geekcafe.com>
 Classifier: License :: Other/Proprietary License

{boto3_assist-0.2.2 → boto3_assist-0.2.3}/pyproject.toml

@@ -7,7 +7,7 @@ packages = ["src/boto3_assist"]
 
 [project]
 name = "boto3_assist"
-version = "0.2.2"
+version = "0.2.3"
 authors = [
   { name="Eric Wilson", email="boto3-assist@geekcafe.com" }
 ]

{boto3_assist-0.2.2 → boto3_assist-0.2.3}/src/boto3_assist/s3/s3.py

@@ -139,7 +139,7 @@ class S3(S3Connection):
 
         return response
 
-    def upload_file_obj(self, bucket: str, key: str, file_obj: bytes) -> str:
+    def upload_file_obj(self, bucket: str, key: str, file_obj: bytes | str) -> str:
         """
         Uploads a file object to s3. Returns the full s3 path s3://<bucket>/<key>
         """
@@ -156,6 +156,10 @@ class S3(S3Connection):
             }
         )
         try:
+            # convert if necessary
+            file_obj: bytes = (
+                file_obj.encode("utf-8") if isinstance(file_obj, str) else file_obj
+            )
             self.client.upload_fileobj(Fileobj=file_obj, Bucket=bucket, Key=key)
 
         except ClientError as ce:
@@ -503,3 +507,19 @@ class S3(S3Connection):
         else:
             # Not in AWS Lambda, use the system's default temp directory
             return tempfile.gettempdir()
+
+    def encode(
+        self, text: str, encoding: str = "utf-8", errors: str = "strict"
+    ) -> bytes:
+        """
+        Encodes a string for s3
+        """
+        return text.encode(encoding=encoding, errors=errors)
+
+    def decode(
+        self, file_obj: bytes, encoding: str = "utf-8", errors: str = "strict"
+    ) -> str:
+        """
+        Decodes bytes to a string
+        """
+        return file_obj.decode(encoding=encoding, errors=errors)

boto3_assist-0.2.3/src/boto3_assist/version.py

@@ -0,0 +1 @@
+__version__ = '0.2.3'

boto3_assist-0.2.2/src/boto3_assist/version.py

@@ -1 +0,0 @@
-__version__ = '0.2.2'