openscap_report.report_generators.json

Complexity

Total Complexity

5

Size/Duplication

Total Lines	41
Duplicated Lines	0 %

Test Coverage

Coverage

36.67%

Importance

Changes

0

3 Methods

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

# 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):
        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("base.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())


class JSONEverythingReportGenerator(ReportGenerator):
    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())