From d5518a994967a91a2df420c72e4a6de1827839ea Mon Sep 17 00:00:00 2001 From: Sergio Date: Tue, 2 Apr 2024 14:21:29 +0200 Subject: [PATCH] fix other providers --- prowler/__main__.py | 2 +- prowler/lib/banner.py | 2 +- prowler/lib/check/checks_loader.py | 2 +- prowler/providers/common/models.py | 4 ++-- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/prowler/__main__.py b/prowler/__main__.py index 12a38fc46b..4fc3ea8e8a 100644 --- a/prowler/__main__.py +++ b/prowler/__main__.py @@ -212,7 +212,7 @@ def prowler(): ) # Prowler Fixer - if args.fix: + if getattr(args, "fix", None): print(f"{Style.BRIGHT}\nRunning Prowler Fixer, please wait...{Style.RESET_ALL}") # Check if there are any FAIL findings if any("FAIL" in finding.status for finding in findings): diff --git a/prowler/lib/banner.py b/prowler/lib/banner.py index 1a45c9a5c2..99fbd4a88d 100644 --- a/prowler/lib/banner.py +++ b/prowler/lib/banner.py @@ -15,7 +15,7 @@ def print_banner(args): """ print(banner) - if args.verbose or args.fix: + if args.verbose or getattr(args, "fix", False): print( f""" Color code for results: diff --git a/prowler/lib/check/checks_loader.py b/prowler/lib/check/checks_loader.py index 4c290df161..2091e64561 100644 --- a/prowler/lib/check/checks_loader.py +++ b/prowler/lib/check/checks_loader.py @@ -101,7 +101,7 @@ def load_checks_to_execute( # Only execute threat detection checks if threat-detection category is set if "threat-detection" not in categories: - for threat_detection_check in check_categories["threat-detection"]: + for threat_detection_check in check_categories.get("threat-detection", []): checks_to_execute.discard(threat_detection_check) # Check Aliases diff --git a/prowler/providers/common/models.py b/prowler/providers/common/models.py index b7f6a23d97..5073219012 100644 --- a/prowler/providers/common/models.py +++ b/prowler/providers/common/models.py @@ -35,14 +35,14 @@ def __init__(self, arguments, bulk_checks_metadata): self.only_logs = arguments.only_logs self.unix_timestamp = arguments.unix_timestamp self.shodan_api_key = arguments.shodan - self.fix = arguments.fix + self.fix = getattr(arguments, "fix", None) # Shodan API Key if arguments.shodan: update_provider_config("shodan_api_key", arguments.shodan) # Check output directory, if it is not created -> create it - if arguments.output_directory and not arguments.fix: + if arguments.output_directory and not self.fix: if not isdir(arguments.output_directory): if arguments.output_formats: makedirs(arguments.output_directory, exist_ok=True)