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(custom checks): fix import from s3 (#2901)

Browse Source
This commit is contained in:
Nacho Rivera
2023-10-03 11:31:55 +02:00
committed by GitHub
parent 0745a57f52
commit bbce2c5e35
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -134,11 +134,13 @@ def parse_checks_from_folder(audit_info, input_folder: str, provider: str) -> in
): ):
bucket = input_folder.split("/")[2] bucket = input_folder.split("/")[2]
key = ("/").join(input_folder.split("/")[3:]) key = ("/").join(input_folder.split("/")[3:])
s3_reource = audit_info.audit_session.resource("s3") s3_resource = audit_info.audit_session.resource("s3")
bucket = s3_reource.Bucket(bucket) bucket = s3_resource.Bucket(bucket)
for obj in bucket.objects.filter(Prefix=key): for obj in bucket.objects.filter(Prefix=key):
if not os.path.exists(os.path.dirname(obj.key)): if not os.path.exists(os.path.dirname(obj.key)):
os.makedirs(os.path.dirname(obj.key)) os.makedirs(os.path.dirname(obj.key))
if obj.key[-1] == "/":
continue
bucket.download_file(obj.key, obj.key) bucket.download_file(obj.key, obj.key)
input_folder = key input_folder = key
# Import custom checks by moving the checks folders to the corresponding services # Import custom checks by moving the checks folders to the corresponding services