diff --git a/main.py b/main.py index c38008f..5e3bf1e 100644 --- a/main.py +++ b/main.py @@ -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