From 0cbe80d2abb7e211bfd9f7fad99a9ca10c03cad3 Mon Sep 17 00:00:00 2001 From: Nacho Rivera <59198746+n4ch04@users.noreply.github.com> Date: Tue, 17 Jan 2023 10:00:31 +0100 Subject: [PATCH] feat(report): conditional import (#1702) Co-authored-by: Pepe Fagoaga --- prowler/lib/check/check.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/prowler/lib/check/check.py b/prowler/lib/check/check.py index a262d7a6..d748b588 100644 --- a/prowler/lib/check/check.py +++ b/prowler/lib/check/check.py @@ -12,7 +12,16 @@ from prowler.config.config import orange_color from prowler.lib.check.compliance_models import load_compliance_framework from prowler.lib.check.models import Check, load_check_metadata from prowler.lib.logger import logger -from prowler.lib.outputs.outputs import report + +try: + lib = os.environ["PROWLER_REPORT_LIB_PATH"] + outputs_module = importlib.import_module(lib) + report = getattr(outputs_module, "report") +except KeyError: + from prowler.lib.outputs.outputs import report +except Exception: + sys.exit() + from prowler.lib.utils.utils import open_file, parse_json_file from prowler.providers.aws.lib.audit_info.models import AWS_Audit_Info from prowler.providers.common.outputs import Provider_Output_Options