ghndrx/prowler
1
0
Fork 0
mirror of https://github.com/ghndrx/prowler.git synced 2026-02-10 14:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(remove_custom_checks_module): delete service folder if empty (#2885)

Browse Source
This commit is contained in:
Nacho Rivera
2023-10-03 10:33:06 +02:00
committed by GitHub
parent e701aca64b
commit 3c396e76f6
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
prowler/lib/check/check.py
View File

@@ -179,6 +179,10 @@ def remove_custom_checks_module(input_folder: str, provider: str):
prowler_module = f"{prowler_dir[0]}/providers/{provider}/services/{check_service}/{check.name}"
if os.path.exists(prowler_module):
shutil.rmtree(prowler_module)
# test if the service only had the loaded custom checks to delete the folder
prowler_service_module = prowler_module.rsplit("/", 1)[0]
if not os.listdir(prowler_service_module):
shutil.rmtree(prowler_service_module)
# If S3 URI, remove the downloaded folders
if s3_uri and os.path.exists(input_folder):
shutil.rmtree(input_folder)