Add config print

Author: ZzEeKkAa

dpbench/console/_namespace.py

@@ -29,6 +29,7 @@ class Namespace(argparse.Namespace):
     timeout: float
     precision: Union[str, None]
     program: str
+    color: str
     comparisons: list[str]
 
 

dpbench/console/config.py

@@ -0,0 +1,61 @@
+# SPDX-FileCopyrightText: 2022 - 2023 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+"""Report subcommand package."""
+
+import argparse
+
+from ._namespace import Namespace
+
+
+def add_config_arguments(parser: argparse.ArgumentParser):
+    """Add arguments for the config subcommand.
+
+    Args:
+        parser: argument parser where arguments will be populated.
+    """
+    parser.add_argument(
+        "--color",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+        help="Set if print with color.",
+    )
+    pass
+
+
+def execute_config(args: Namespace):
+    """Execute config sub command.
+
+    Print loaded configuration in pretty format
+
+    Args:
+        args: object with all input arguments.
+        conn: database connection.
+    """
+    import dpbench.config as cfg
+
+    cfg.GLOBAL = cfg.read_configs(
+        benchmarks=args.benchmarks,
+        implementations=args.implementations,
+        with_npbench=True,
+        with_polybench=True,
+    )
+
+    if args.color:
+        from pprint import pformat
+
+        from pygments import highlight
+        from pygments.formatters import Terminal256Formatter
+        from pygments.lexers import PythonLexer
+
+        print(
+            highlight(
+                pformat(cfg.GLOBAL), PythonLexer(), Terminal256Formatter()
+            ),
+            end="",
+        )
+    else:
+        from pprint import pprint
+
+        pprint(cfg.GLOBAL)

dpbench/console/entry.py

@@ -8,6 +8,7 @@
 from importlib.metadata import version
 
 from ._namespace import CommaSeparateStringAction, Namespace
+from .config import add_config_arguments, execute_config
 from .report import add_report_arguments, execute_report
 from .run import add_run_arguments, execute_run
 
@@ -85,6 +86,13 @@ def parse_args() -> Namespace:
 
     add_report_arguments(report_parser)
 
+    config_parser = subparsers.add_parser(
+        "config",
+        description="Subcommand to print config.",
+    )
+
+    add_config_arguments(config_parser)
+
     return parser.parse_args(namespace=Namespace())
 
 
@@ -110,6 +118,8 @@ def main():
         execute_run(args, conn)
     elif args.program == "report":
         execute_report(args, conn)
+    elif args.program == "config":
+        execute_config(args)
 
 
 if __name__ == "__main__":