From 082a0ae3f6131e5d1725ee7c9aaad2b3d3628c2a Mon Sep 17 00:00:00 2001
From: Vu Dao <mr.minhvu.dao@gmail.com>
Date: Sat, 1 Feb 2025 00:07:51 +0700
Subject: [PATCH] fix: Delete download path after scanning finished

---
 assets/lambda/code/scan/lambda.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/assets/lambda/code/scan/lambda.py b/assets/lambda/code/scan/lambda.py
index 845629d7..df620e41 100644
--- 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"""