feat(checks): Select checks to run from provider using -c/--checks (#1197)

* feat(checks): Select checks to run

* Update providers/aws/services/iam/iam_disable_30_days_credentials/iam_disable_30_days_credentials.py

Co-authored-by: Sergio Garcia <38561120+sergargar@users.noreply.github.com>

prowler.py

@@ -2,44 +2,17 @@
 # -*- coding: utf-8 -*-
 
 import argparse
-import importlib
-import pkgutil
 
 from lib.banner import print_banner, print_version
+from lib.check import import_check, load_checks_to_execute, run_check
 from lib.logger import logger, logging_levels
-from lib.outputs import report
 from providers.aws.aws_provider import Input_Data, provider_set_session
 
-
-def run_check(check):
-    print(f"\nCheck Name: {check.CheckName}")
-    findings = check.execute()
-    report(findings)
-
-
-def import_check(check_path):
-    lib = importlib.import_module(f"{check_path}")
-    return lib
-
-
-def recover_modules_from_provider(provider):
-    modules = []
-    for module_name in pkgutil.walk_packages(
-        importlib.import_module(f"providers.{provider}.services").__path__,
-        importlib.import_module(f"providers.{provider}.services").__name__ + ".",
-    ):
-        if module_name.name.count(".") == 5:
-            modules.append(module_name.name)
-    return modules
-
-
 if __name__ == "__main__":
-    # start_time = time.time()
+    # CLI Arguments
     parser = argparse.ArgumentParser()
-    parser.add_argument("provider", help="Specify Provider: AWS")
-    parser.add_argument(
-        "-c", "--checks", nargs="+", help="Comma separated list of checks"
-    )
+    parser.add_argument("provider", choices=["aws"], help="Specify Provider")
+    parser.add_argument("-c", "--checks", nargs="+", help="List of checks")
     parser.add_argument(
         "-b", "--no-banner", action="store_false", help="Hide Prowler Banner"
     )
@@ -49,7 +22,7 @@ if __name__ == "__main__":
     parser.add_argument(
         "--log-level",
         choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
-        default="CRITICAL",
+        default="ERROR",
         help="Select Log Level",
     )
     parser.add_argument(
@@ -135,33 +108,28 @@ if __name__ == "__main__":
     # Setting session
     provider_set_session(session_input)
 
-    # libreria para generar la lista de checks
-    if checks:
-        for check in checks:
-            # Recover service from check name
-            service = check.split("_")[0]
-            # Import check module
-            lib = import_check(
-                f"providers.{provider}.services.{service}.{check}.{check}"
-            )
-            # Recover functions from check
-            check_to_execute = getattr(lib, check)
-            c = check_to_execute()
-            # Run check
-            run_check(c)
+    # Load checks to execute
+    logger.debug("Loading checks")
+    checks_to_execute = load_checks_to_execute(checks, provider)
 
-    else:
-        # Get all check modules to run
-        modules = recover_modules_from_provider(provider)
-        # Run checks
-        for check_module in modules:
-            print(check_module)
+    # Execute checks
+    for check_name in checks_to_execute:
+        # Recover service from check name
+        service = check_name.split("_")[0]
+        try:
             # Import check module
-            lib = import_check(check_module)
-            # Recover module from check name
-            check_name = check_module.split(".")[5]
+            check_module_path = (
+                f"providers.{provider}.services.{service}.{check_name}.{check_name}"
+            )
+            lib = import_check(check_module_path)
             # Recover functions from check
             check_to_execute = getattr(lib, check_name)
             c = check_to_execute()
             # Run check
             run_check(c)
+
+        # If check does not exists in the provider or is from another provider
+        except ModuleNotFoundError:
+            logger.error(
+                f"Check '{check_name}' was not found for the {provider.upper()} provider"
+            )