Update main.py

2024-12-12 22:34:20 +01:00
parent 010adf6f39
commit 3d52ad7544
1 changed files with 2 additions and 1 deletions
@@ -54,7 +54,8 @@ def retrieve_file_contents(path, pPullFromS3):
            "content": content
        })

-    shutil.rmtree(pulled_bucket_data)
+    if os.path.exists(pulled_bucket_data):
+        shutil.rmtree(pulled_bucket_data)

    return contents