1
|
|
|
''' |
2
|
|
|
This file contains a class for command arf-to-graph |
3
|
|
|
''' |
4
|
|
|
|
5
|
|
|
from .client import Client |
6
|
|
|
from .converter import Converter |
7
|
|
|
from ._builder_html_graph import BuilderHtmlGraph |
8
|
|
|
from .xml_parser import XmlParser |
9
|
|
|
|
10
|
|
|
|
11
|
|
|
class ArfToHtml(Client): |
|
|
|
|
12
|
|
|
def _set_attributes(self): |
13
|
|
|
self.all_in_one = self.arg.all_in_one |
14
|
|
|
self.all_rules = True if self.all_in_one else self.arg.all |
15
|
|
|
self.display_html = True if self.out is None else self.arg.display |
16
|
|
|
self.show_failed_rules = self.arg.show_failed_rules |
17
|
|
|
self.show_not_selected_rules = self.arg.show_not_selected_rules |
18
|
|
|
self.html_builder = BuilderHtmlGraph(self.parts, self.verbose) |
|
|
|
|
19
|
|
|
self.xml_parser = XmlParser(self.source_filename) |
20
|
|
|
|
21
|
|
|
def _get_message(self): |
22
|
|
|
MESSAGES = { |
|
|
|
|
23
|
|
|
'description': 'Client for visualization of SCAP rule evaluation results', |
24
|
|
|
'source_filename': 'ARF scan file', |
25
|
|
|
} |
26
|
|
|
return MESSAGES |
27
|
|
|
|
28
|
|
|
def create_dict_of_rule(self, rule_id): |
|
|
|
|
29
|
|
|
converter = Converter(self.xml_parser.get_oval_tree(rule_id)) |
30
|
|
|
return converter.to_JsTree_dict(self.hide_passing_tests) |
31
|
|
|
|
32
|
|
|
def _put_to_dict_oval_trees(self, dict_oval_trees, rule): |
33
|
|
|
dict_oval_trees[rule] = self.create_dict_of_rule(rule) |
34
|
|
|
|
35
|
|
|
def _get_src_for_one_graph(self, rule): |
36
|
|
|
return self.get_save_src(rule + self.date) |
37
|
|
|
|
38
|
|
|
def prepare_parser(self): |
39
|
|
|
super().prepare_parser() |
40
|
|
|
self.prepare_args_when_output_is_html() |
41
|
|
|
self.prepare_args_when_user_can_list_in_rules() |
42
|
|
|
|