mirror of
https://github.com/ghndrx/prowler.git
synced 2026-02-10 14:55:00 +00:00
feat(quiet): Add -q option. (#1211)
Co-authored-by: sergargar <sergio@verica.io>
This commit is contained in:
@@ -3,6 +3,7 @@ import pkgutil
|
||||
from abc import ABC, abstractmethod
|
||||
from dataclasses import dataclass
|
||||
from types import ModuleType
|
||||
|
||||
from colorama import Fore, Style
|
||||
|
||||
from config.config import groups_file
|
||||
@@ -158,11 +159,22 @@ def recover_modules_from_provider(provider: str, service: str = None) -> list:
|
||||
return modules
|
||||
|
||||
|
||||
def set_output_options(quiet):
|
||||
global output_options
|
||||
output_options = Output_From_Options(
|
||||
is_quiet=quiet
|
||||
# set input options here
|
||||
)
|
||||
return output_options
|
||||
|
||||
|
||||
def run_check(check):
|
||||
print(f"\nCheck Name: {check.CheckName} - {Fore.MAGENTA}{check.ServiceName}{Fore.YELLOW}[{check.Severity}]{Style.RESET_ALL}")
|
||||
print(
|
||||
f"\nCheck Name: {check.CheckName} - {Fore.MAGENTA}{check.ServiceName}{Fore.YELLOW}[{check.Severity}]{Style.RESET_ALL}"
|
||||
)
|
||||
logger.debug(f"Executing check: {check.CheckName}")
|
||||
findings = check.execute()
|
||||
report(findings)
|
||||
report(findings, output_options)
|
||||
|
||||
|
||||
def import_check(check_path: str) -> ModuleType:
|
||||
@@ -182,6 +194,11 @@ class Check_Report:
|
||||
self.result_extended = ""
|
||||
|
||||
|
||||
@dataclass
|
||||
class Output_From_Options:
|
||||
is_quiet: bool
|
||||
|
||||
|
||||
class Check(ABC):
|
||||
def __init__(self):
|
||||
try:
|
||||
|
||||
@@ -1,13 +1,18 @@
|
||||
from colorama import Fore, Style
|
||||
|
||||
|
||||
def report(check_findings):
|
||||
def report(check_findings, output_options):
|
||||
check_findings.sort(key=lambda x: x.region)
|
||||
for finding in check_findings:
|
||||
color = set_report_color(finding.status)
|
||||
print(
|
||||
f"{color}{finding.status}{Style.RESET_ALL} {finding.region}: {finding.result_extended}"
|
||||
)
|
||||
if output_options.is_quiet and "FAIL" in finding.status:
|
||||
print(
|
||||
f"{color}{finding.status}{Style.RESET_ALL} {finding.region}: {finding.result_extended}"
|
||||
)
|
||||
elif not output_options.is_quiet:
|
||||
print(
|
||||
f"{color}{finding.status}{Style.RESET_ALL} {finding.region}: {finding.result_extended}"
|
||||
)
|
||||
|
||||
|
||||
def set_report_color(status):
|
||||
|
||||
Reference in New Issue
Block a user