openscap_report.scap_results_parser.data_structures.remediation

Complexity

Total Complexity

2

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Test Coverage

Coverage

93.75%

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	Remediation.get_type()	0	11	1
A	Remediation.as_dict()	0	2	1

# Copyright 2022, Red Hat, Inc.
# SPDX-License-Identifier: LGPL-2.1-or-later

from openscap_report.dataclasses import asdict, dataclass

REMEDIATION_JSON_KEYS = [
    "remediation_id",
    "system",
    "complexity",
    "disruption",
    "strategy",
    "fix",
]

HIDDEN_REMEDIATION_TYPES = [
    "urn:redhat:anaconda:pre",
    "urn:xccdf:fix:script:kickstart",
]


@dataclass
class Remediation:
    remediation_id: str
    system: str = ""
    complexity: str = ""
    disruption: str = ""
    strategy: str = ""
    fix: str = ""

    def as_dict(self):
        return asdict(self)

    def get_type(self):
        script_types = {
            "urn:xccdf:fix:script:sh": "Shell script",
            "urn:xccdf:fix:script:ansible": "Ansible snippet",
            "urn:xccdf:fix:script:puppet": "Puppet snippet",
            "urn:redhat:anaconda:pre": "Anaconda snippet",
            "urn:xccdf:fix:script:kubernetes": "Kubernetes snippet",
            "urn:redhat:osbuild:blueprint": "OSBuild Blueprint snippet",
            "urn:xccdf:fix:script:kickstart": "Kickstart snippet",
        }
        return script_types.get(self.system, "script")