From b33b529e74b8d3dff7e6bf886f2e93b3e9c18981 Mon Sep 17 00:00:00 2001
From: Pepe Fagoaga <pepe@prowler.com>
Date: Mon, 3 Jun 2024 14:44:14 +0200
Subject: [PATCH] refactor(banner): remove unneeded arguments (#4155)

---
 dashboard/__main__.py |  2 +-
 prowler/__main__.py   |  3 ++-
 prowler/lib/banner.py | 13 +++++++++++--
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/dashboard/__main__.py b/dashboard/__main__.py
index 4454714c0b..b82e7c62f7 100644
--- a/dashboard/__main__.py
+++ b/dashboard/__main__.py
@@ -16,7 +16,7 @@ from prowler.lib.banner import print_banner
 warnings.filterwarnings("ignore")
 
 cli = sys.modules["flask.cli"]
-print_banner(verbose=False)
+print_banner()
 print(
     f"{Fore.GREEN}Loading all CSV files from the folder {folder_path_overview} ...\n{Style.RESET_ALL}"
 )
diff --git a/prowler/__main__.py b/prowler/__main__.py
index 5fb96602aa..060c264acc 100644
--- a/prowler/__main__.py
+++ b/prowler/__main__.py
@@ -89,7 +89,8 @@ def prowler():
     )
 
     if not args.no_banner:
-        print_banner(args.verbose, getattr(args, "fixer", None))
+        legend = args.verbose or getattr(args, "fixer", None)
+        print_banner(legend)
 
     # We treat the compliance framework as another output format
     if compliance_framework:
diff --git a/prowler/lib/banner.py b/prowler/lib/banner.py
index 48cbc3bffa..690137be20 100644
--- a/prowler/lib/banner.py
+++ b/prowler/lib/banner.py
@@ -3,7 +3,16 @@ from colorama import Fore, Style
 from prowler.config.config import banner_color, orange_color, prowler_version, timestamp
 
 
-def print_banner(verbose: bool, fixer: bool = False):
+def print_banner(legend: bool = False):
+    """
+    Prints the banner with optional legend for color codes.
+
+    Parameters:
+    - legend (bool): Flag to indicate whether to print the color legend or not. Default is False.
+
+    Returns:
+    - None
+    """
     banner = rf"""{banner_color}                         _
  _ __  _ __ _____      _| | ___ _ __
 | '_ \| '__/ _ \ \ /\ / / |/ _ \ '__|
@@ -15,7 +24,7 @@ def print_banner(verbose: bool, fixer: bool = False):
 """
     print(banner)
 
-    if verbose or fixer:
+    if legend:
         print(
             f"""
 {Style.BRIGHT}Color code for results:{Style.RESET_ALL}