Add JSON mini version of output

OpenSCAP · Aug 17, 2022 · 0f798bc · 0f798bc
1 parent 405610a
commit 0f798bc
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/openscap_report/cli.py b/openscap_report/cli.py
@@ -10,7 +10,8 @@
 
 from . import __version__
 from .debug_settings import DebugSetting
-from .report_generators import (HTMLReportGenerator, JSONReportGenerator,
+from .report_generators import (HTMLReportGenerator, JSONMiniReportGenerator,
+                                JSONReportGenerator,
                                 OldStyleHTMLReportGenerator)
 from .scap_results_parser import SCAPResultsParser
 
@@ -93,7 +94,7 @@ def prepare_parser():
         "--format",
         action="store",
         default="HTML",
-        choices=["HTML", "OLD-STYLE-HTML", "JSON"],
+        choices=["HTML", "OLD-STYLE-HTML", "JSON", "JSON-MINI"],
         help="FORMAT: %(choices)s (default: %(default)s)."
     )
     parser.add_argument(
@@ -141,6 +142,7 @@ def get_report_generator(self, report_parser):
             "HTML": HTMLReportGenerator,
             "OLD-STYLE-HTML": OldStyleHTMLReportGenerator,
             "JSON": JSONReportGenerator,
+            "JSON-MINI": JSONMiniReportGenerator,
         }
         return dict_of_report_generators[self.output_format](report_parser)
 

diff --git a/openscap_report/report_generators/__init__.py b/openscap_report/report_generators/__init__.py
@@ -3,6 +3,6 @@
 
 from .exceptions import FilterNotSupportDataStructureException
 from .html import HTMLReportGenerator
-from .json import JSONReportGenerator
+from .json import JSONMiniReportGenerator, JSONReportGenerator
 from .old_style_html import OldStyleHTMLReportGenerator
 from .report_generator import ReportGenerator
diff --git a/openscap_report/report_generators/json.py b/openscap_report/report_generators/json.py
@@ -3,6 +3,9 @@
 import json
 import logging
 from io import BytesIO
+from pathlib import Path
+
+from jinja2 import Environment, FileSystemLoader
 
 from .report_generator import ReportGenerator
 
@@ -19,3 +22,22 @@ def generate_report(self, debug_setting):
             indent = "\t"
         json_data = json.dumps(self.report.as_dict(), indent=indent)
         return BytesIO(json_data.encode())
+
+
+class JSONMiniReportGenerator(JSONReportGenerator):
+    def __init__(self, parser):
+        super().__init__(parser)
+        self.file_loader = FileSystemLoader(str(Path(__file__).parent / "json_templates"))
+        self.env = Environment(loader=self.file_loader)
+        self.env.trim_blocks = True
+        self.env.lstrip_blocks = True
+
+    def generate_report(self, debug_setting):
+        logging.warning("JSON Format is experimental output!")
+        template = self.env.get_template("mini.json")
+        json_data = json.loads(template.render(report=self.report, debug_setting=debug_setting))
+        indent = None
+        if debug_setting.no_minify:
+            indent = "\t"
+        json_report = json.dumps(json_data, indent=indent)
+        return BytesIO(json_report.encode())