fix: Delete download path after scanning finished

awslabs · Jan 31, 2025 · 082a0ae · 082a0ae
1 parent 5151595
commit 082a0ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/assets/lambda/code/scan/lambda.py b/assets/lambda/code/scan/lambda.py
@@ -315,7 +315,8 @@ def delete(download_path, input_key=None):
                 shutil.rmtree(obj)
             else:
                 os.remove(obj)
-
+        # Remove the download_path folder itself
+        shutil.rmtree(download_path)
 
 def report_failure(input_bucket, input_key, download_path, message):
     """Set the S3 object tag to ERROR if scan function fails"""