import StringIO and pass data through StringIO read (#173)

ysde · Apr 13, 2023 · 843ffe9 · 843ffe9 · spyder007 · Apr 17, 2023
1 parent 3afb563
commit 843ffe9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/grafana_backup/s3_download.py b/grafana_backup/s3_download.py
@@ -1,5 +1,6 @@
 import boto3
 from botocore.exceptions import NoCredentialsError, ClientError
+import StringIO
 
 
 def main(args, settings):
@@ -22,7 +23,8 @@ def main(args, settings):
 
     try:
         # .read() left off on purpose, tarfile.open() takes care of that part
-        s3_data = s3_object.get()["Body"]
+        s3_data_raw = s3_object.get()["Body"]
+        s3_data = StringIO.StringIO(s3_data_raw.read())
         print("Download from S3 was successful")
     except ClientError as e:
         if e.response["Error"]["Code"] == "NoSuchKey":