From d3f19053f2a303184135fd18c34d51d25802aa1b Mon Sep 17 00:00:00 2001 From: greg Date: Tue, 12 Mar 2024 02:15:08 -0700 Subject: [PATCH] add content type pdf/html for uploads to s3 --- prowler/providers/aws/lib/s3/s3.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/prowler/providers/aws/lib/s3/s3.py b/prowler/providers/aws/lib/s3/s3.py index 6ebb711f..f94d7c63 100644 --- a/prowler/providers/aws/lib/s3/s3.py +++ b/prowler/providers/aws/lib/s3/s3.py @@ -36,7 +36,14 @@ def send_to_s3_bucket( object_name = bucket_directory + "/" + output_mode + "/" + filename 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: 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 bucket_remote_dir = bucket_remote_dir.partition("prowler/")[-1] - return bucket_remote_dir + return bucket_remote_dir \ No newline at end of file