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

add content type pdf/html for uploads to s3

Browse Source
This commit is contained in:
greg
2024-03-12 02:15:08 -07:00
parent 8242a882a3
commit d3f19053f2
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
prowler/providers/aws/lib/s3/s3.py
View File

@@ -36,7 +36,14 @@ def send_to_s3_bucket(
object_name = bucket_directory + "/" + output_mode + "/" + filename object_name = bucket_directory + "/" + output_mode + "/" + filename
s3_client = audit_session.client("s3") s3_client = audit_session.client("s3")
s3_client.upload_file(file_name, output_bucket_name, object_name)
extra_args = {}
if filename.endswith('.html'):
extra_args['ContentType'] = 'text/html'
elif filename.endswith('.pdf'):
extra_args['ContentType'] = 'application/pdf'
s3_client.upload_file(file_name, output_bucket_name, object_name, ExtraArgs=extra_args)
except Exception as error: except Exception as error:
logger.error( logger.error(
@@ -49,4 +56,4 @@ def get_s3_object_path(output_directory: str) -> str:
if "prowler/" in bucket_remote_dir: # Check if it is not a custom directory if "prowler/" in bucket_remote_dir: # Check if it is not a custom directory
bucket_remote_dir = bucket_remote_dir.partition("prowler/")[-1] bucket_remote_dir = bucket_remote_dir.partition("prowler/")[-1]
return bucket_remote_dir return bucket_remote_dir