openscap_report.scap_results_parser.data_structures.profile_info

Complexity

Total Complexity

9

Size/Duplication

Total Lines	48
Duplicated Lines	0 %

Test Coverage

Coverage

81.48%

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	ProfileInfo.select_rules()	0	3	2
A	ProfileInfo.get_cpe_platforms_that_satisfy_evaluation_target()	0	2	1
A	ProfileInfo.get_list_of_cpe_platforms_that_satisfy_evaluation_target()	0	3	1
A	ProfileInfo.deselect_rules()	0	4	3
A	ProfileInfo.as_dict()	0	2	1
A	ProfileInfo.get_applicable_cpe_platforms_for_profile()	0	2	1

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

from typing import Dict, List

from openscap_report.dataclasses import asdict, dataclass, field

PROFILE_JSON_KEYS = [
    "profile_id",
    "description",
    "title",
    "extends",
]


@dataclass
class ProfileInfo:
    profile_id: str
    description: str
    title: str
    extends: str = None
    cpe_platforms_for_profile: Dict[str, bool] = field(default_factory=dict)
    selected_rules_ids: List[str] = field(default_factory=list)
    selected_groups_ids: List[str] = field(default_factory=list)

    def as_dict(self):
        return asdict(self)

    def get_applicable_cpe_platforms_for_profile(self):
        return ", ".join(self.cpe_platforms_for_profile.keys())

    def get_cpe_platforms_that_satisfy_evaluation_target(self):
        return ", ".join(self.get_list_of_cpe_platforms_that_satisfy_evaluation_target())

    def get_list_of_cpe_platforms_that_satisfy_evaluation_target(self):
        return [
            cpe_id for cpe_id, is_satisfy in self.cpe_platforms_for_profile.items() if is_satisfy
        ]

    def deselect_rules(self, rule_ids):
        for rule_id in rule_ids:
            if rule_id in self.selected_rules_ids:
                self.selected_rules_ids.remove(rule_id)

    def select_rules(self, rule_ids):
        if len(self.selected_rules_ids) > 0:
            self.selected_rules_ids.extend(rule_ids)