feat(severity): Run checks by severity (#1223)

This commit is contained in:
Pepe Fagoaga
2022-06-23 16:56:06 +02:00
committed by GitHub
parent 2288702d26
commit 66d57a3d36
3 changed files with 41 additions and 18 deletions

View File

@@ -98,6 +98,20 @@ def print_services(service_list: set):
print(f"- {service}")
def print_checks(provider: str, check_list: set, bulk_checks_metadata: dict):
for check in check_list:
try:
print(
f"[{bulk_checks_metadata[check].CheckID}] {bulk_checks_metadata[check].CheckTitle} - {Fore.MAGENTA}{bulk_checks_metadata[check].ServiceName} {Fore.YELLOW}[{bulk_checks_metadata[check].Severity}]{Style.RESET_ALL}"
)
except KeyError as error:
logger.error(
f"Check {error} was not found for the {provider.upper()} provider"
)
# List available groups
def list_groups(provider: str):
groups = parse_groups_from_file(groups_file)

View File

@@ -15,6 +15,7 @@ def load_checks_to_execute(
check_list: list,
service_list: list,
group_list: list,
severities: list,
provider: str,
) -> set:
@@ -25,10 +26,13 @@ def load_checks_to_execute(
for check_name in check_list:
checks_to_execute.add(check_name)
# elif severity_list:
# using bulk_checks_metadata
# elif compliance_list:
# using bulk_checks_metadata
# Handle if there are some severities passed using --severity
elif severities:
for check in bulk_checks_metadata:
# Check check's severity
if bulk_checks_metadata[check].Severity in severities:
checks_to_execute.add(check)
# Handle if there are checks passed using -C/--checks-file
elif checks_file:
try: