openscap_report.report_generators.json

Complexity

Total Complexity

6

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Test Coverage

Coverage

37.5%

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	JSONMiniReportGenerator.generate_report()	0	9	2
A	JSONReportGenerator.generate_report()	0	8	2
A	JSONReportGenerator.__init__()	0	2	1
A	JSONMiniReportGenerator.__init__()	0	6	1

# Copyright 2022, Red Hat, Inc.
# SPDX-License-Identifier: LGPL-2.1-or-later
import json
import logging
from io import BytesIO
from pathlib import Path

from jinja2 import Environment, FileSystemLoader

from .report_generator import ReportGenerator


class JSONReportGenerator(ReportGenerator):
    def __init__(self, parser):  # pylint: disable=W0231
        self.report = parser.parse_report()

    def generate_report(self, debug_setting):
        logging.warning("JSON Format is experimental output!")

        indent = None
        if debug_setting.no_minify:
            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())