oval_graph.client.Client._get_wanted_not_selected_rules() - Code Metrics - Inspection of "Merge pull request #76 from Honny1/optionShowOnlyF..." - OpenSCAP/oval-graph - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( be999b...2f9937 )

by Jan

created 2020-01-22 16:04 UTC

Client._get_wanted_not_selected_rules() A

↳ Parent: oval_graph.client

Complexity

Conditions

Size

Total Lines	4
Code Lines	4

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	2
CRAP Score	1

Importance

Changes

Metric	Value
cc	1
eloc	4
nop	1
dl	0
loc	4
ccs	2
cts	2
cp	1
crap	1
rs	10
c	0
b	0
f	0

import re
import argparse
import tempfile
import os
import webbrowser
import json
import shutil
from datetime import datetime
import sys

from .xml_parser import XmlParser


class Client():
    def __init__(self, args):
        self.parser = None
        self.MESSAGES = self._get_message()
        self.arg = self.parse_arguments(args)
        self.hide_passing_tests = self.arg.hide_passing_tests
        self.source_filename = self.arg.source_filename
        self.rule_name = self.arg.rule_id
        self.out = self.arg.output
        self.all_rules = self.arg.all
        self.isatty = sys.stdout.isatty()
        self.show_fail_rules = False
        self.show_not_selected_rules = False
        self.xml_parser = XmlParser(
            self.source_filename)

    def _get_message(self):
        MESSAGES = {
            'description': '',
            '--output': '',
            'source_filename': '',
        }
        return MESSAGES


    def run_gui_and_return_answers(self):
        if self.isatty:
            if self.all_rules:
                return {'rules': [
                    rule['id_rule'] for rule in self.search_rules_id()]}
            else:
                try:
                    import inquirer
                    return inquirer.prompt(self.get_questions())
                except ImportError:
                    print(self.get_selection_rules())
                    return None
        else:
            return {'rules': [
                rule['id_rule'] for rule in self.search_rules_id()]}

    def get_list_of_matched_rules(self):
        rules = self.search_rules_id()
        if self.show_fail_rules:
            rules = self._get_only_fail_rule(rules)
        return rules

    def get_list_of_lines(self):
        lines = ['== The Rule IDs ==']
        for rule in self.get_list_of_matched_rules():
            lines.append("'" + rule['id_rule'] + r'\b' + "'")
        if self.show_not_selected_rules:
            for line in self.get_lines_of_wanted_not_selected_rules():
                lines.append(line)
        lines.append(
            "You haven't got installed inquirer lib. "
            "Please copy id rule with you want use and put it in command")
        return lines

    def get_selection_rules(self):
        return "\n".join(self.get_list_of_lines())

    def get_lines_of_wanted_not_selected_rules(self):
        out = []
        out.append('== The not selected rule IDs ==')
        for rule in self._get_wanted_not_selected_rules():
            out.append(rule['id_rule'] + '(Not selected)')
        return out

    def get_choices(self):
        rules = self.search_rules_id()
        if self.show_fail_rules:
            rules = self._get_only_fail_rule(rules)
        choices = []
        for rule in rules:
            choices.append(rule['id_rule'])
        if self.show_not_selected_rules:
            print("\n".join(self.get_lines_of_wanted_not_selected_rules()))
        return choices

    def get_questions(self):
        choices = self.get_choices()
        from inquirer.questions import Checkbox as checkbox
        questions = [
            checkbox(
                'rules',
                message=(
                    "= The Rules IDs = (move - UP and DOWN arrows,"
                    " select - SPACE or LEFT and RIGHT arrows, submit - ENTER)"),
                choices=choices,
            ),
        ]
        return questions

    def _get_only_fail_rule(self, rules):
        return list(filter(lambda rule: rule['result'] == 'fail', rules))

    def _get_wanted_rules(self):
        return [
            x for x in self.xml_parser.used_rules if re.search(
                self.rule_name, x['id_rule'])]

    def _get_wanted_not_selected_rules(self):
        return [
            x for x in self.xml_parser.notselected_rules if re.search(
                self.rule_name, x['id_rule'])]

    def search_rules_id(self):
        rules = self._get_wanted_rules()
        notselected_rules = self._get_wanted_not_selected_rules()
        if len(notselected_rules) and not rules:
            raise ValueError(
                ('err- rule(s) "{}" was not selected, '
                 "so there are no results. The rule is"
                 ' "notselected" because it'
                 " wasn't a part of the executed profile"
                 " and therefore it wasn't evaluated "
                 "during the scan.")
                .format(notselected_rules[0]['id_rule']))
        elif not notselected_rules and not rules:
            raise ValueError('err- 404 rule not found!')
        else:
            return rules

    def open_web_browser(self, src):
        if not self.off_webbrowser:
            src = os.path.join(src, 'index.html')
            try:
                webbrowser.get('firefox').open_new_tab(src)
            except BaseException:
                webbrowser.open_new_tab(src)

    def copy_interpreter(self, dst):
        src = self.get_src('tree_html_interpreter')
        os.mkdir(dst)
        for item in os.listdir(src):
            s = os.path.join(src, item)
            d = os.path.join(dst, item)
            if os.path.isdir(s):
                shutil.copytree(s, d)
            else:
                shutil.copy2(s, d)

    def get_src(self, src):
        _dir = os.path.dirname(os.path.realpath(__file__))
        FIXTURE_DIR = os.path.join(_dir, src)
        return str(FIXTURE_DIR)

    def save_dict(self, dict_, src):
        with open(os.path.join(src, 'data.js'), "w+") as data_file:
            data_file.write("var data_json =" + str(json.dumps(
                dict_, sort_keys=False, indent=4) + ";"))

    def get_save_src(self, rule):
        date = str(datetime.now().strftime("-%d_%m_%Y-%H_%M_%S"))
        if self.out is not None:
            if not os.path.isdir(self.out):
                os.mkdir(self.out)
                return os.path.join(self.out, 'graph-of-' + rule + date)
            return os.path.join(
                self.out,
                'graph-of-' + rule + date)
        return os.path.join(
            os.getcwd(),
            'graph-of-' + rule + date)

    def parse_arguments(self, args):
        self.prepare_parser()
        args = self.parser.parse_args(args)
        return args

    def prepare_parser(self):
        self.parser = argparse.ArgumentParser(
            description=self.MESSAGES.get('description'))
        self.parser.add_argument(
            '--all',
            action="store_true",
            default=False,
            help="Process all matched rules.")
        self.parser.add_argument(
            '--hide-passing-tests',
            action="store_true",
            default=False,
            help=(
                "Do not display passing tests for better orientation in"
                " graphs that contain a large amount of nodes.(Not implemented)"))
        self.parser.add_argument(
            '-o',
            '--output',
            action="store",
            default=None,
            help=self.MESSAGES.get('--output'))
        self.parser.add_argument(
            "source_filename",
            help=self.MESSAGES.get('source_filename'))
        self.parser.add_argument(
            "rule_id", help=(
                "Rule ID to be visualized. A part from the full rule ID"
                " a part of the ID or a regular expression can be used."
                " If brackets are used in the regular expression "
                "the regular expression must be quoted."))


1	1	import re
2	1	import argparse
3	1	import tempfile
4	1	import os
5	1	import webbrowser
6	1	import json
7	1	import shutil
8	1	from datetime import datetime
9	1	import sys
10
11	1	from .xml_parser import XmlParser
12
13
14	1	class Client():
15	1	def __init__(self, args):
16	1	self.parser = None
17	1	self.MESSAGES = self._get_message()
18	1	self.arg = self.parse_arguments(args)
19	1	self.hide_passing_tests = self.arg.hide_passing_tests
20	1	self.source_filename = self.arg.source_filename
21	1	self.rule_name = self.arg.rule_id
22	1	self.out = self.arg.output
23	1	self.all_rules = self.arg.all
24	1	self.isatty = sys.stdout.isatty()
25	1	self.show_fail_rules = False
26	1	self.show_not_selected_rules = False
27	1	self.xml_parser = XmlParser(
28		self.source_filename)
29
30	1	def _get_message(self):
31	1	MESSAGES = {
32		'description': '',
33		'--output': '',
34		'source_filename': '',
35		}
36	1	return MESSAGES
37
38
39	1	def run_gui_and_return_answers(self):
40	1	if self.isatty:
41	1	if self.all_rules:
42		return {'rules': [
43		rule['id_rule'] for rule in self.search_rules_id()]}
44		else:
45	1	try:
46	1	import inquirer
47		return inquirer.prompt(self.get_questions())
48	1	except ImportError:
49	1	print(self.get_selection_rules())
50	1	return None
51		else:
52		return {'rules': [
53		rule['id_rule'] for rule in self.search_rules_id()]}
54
55	1	def get_list_of_matched_rules(self):
56	1	rules = self.search_rules_id()
57	1	if self.show_fail_rules:
58	1	rules = self._get_only_fail_rule(rules)
59	1	return rules
60
61	1	def get_list_of_lines(self):
62	1	lines = ['== The Rule IDs ==']
63	1	for rule in self.get_list_of_matched_rules():
64	1	lines.append("'" + rule['id_rule'] + r'\b' + "'")
65	1	if self.show_not_selected_rules:
66	1	for line in self.get_lines_of_wanted_not_selected_rules():
67	1	lines.append(line)
68	1	lines.append(
69		"You haven't got installed inquirer lib. "
70		"Please copy id rule with you want use and put it in command")
71	1	return lines
72
73	1	def get_selection_rules(self):
74	1	return "\n".join(self.get_list_of_lines())
75
76	1	def get_lines_of_wanted_not_selected_rules(self):
77	1	out = []
78	1	out.append('== The not selected rule IDs ==')
79	1	for rule in self._get_wanted_not_selected_rules():
80	1	out.append(rule['id_rule'] + '(Not selected)')
81	1	return out
82
83	1	def get_choices(self):
84	1	rules = self.search_rules_id()
85	1	if self.show_fail_rules:
86	1	rules = self._get_only_fail_rule(rules)
87	1	choices = []
88	1	for rule in rules:
89	1	choices.append(rule['id_rule'])
90	1	if self.show_not_selected_rules:
91	1	print("\n".join(self.get_lines_of_wanted_not_selected_rules()))
92	1	return choices
93
94	1	def get_questions(self):
95	1	choices = self.get_choices()
96	1	from inquirer.questions import Checkbox as checkbox
97	1	questions = [
98		checkbox(
99		'rules',
100		message=(
101		"= The Rules IDs = (move - UP and DOWN arrows,"
102		" select - SPACE or LEFT and RIGHT arrows, submit - ENTER)"),
103		choices=choices,
104		),
105		]
106	1	return questions
107
108	1	def _get_only_fail_rule(self, rules):
109	1	return list(filter(lambda rule: rule['result'] == 'fail', rules))
110
111	1	def _get_wanted_rules(self):
112	1	return [
113		x for x in self.xml_parser.used_rules if re.search(
114		self.rule_name, x['id_rule'])]
115
116	1	def _get_wanted_not_selected_rules(self):
117	1	return [
118		x for x in self.xml_parser.notselected_rules if re.search(
119		self.rule_name, x['id_rule'])]
120
121	1	def search_rules_id(self):
122	1	rules = self._get_wanted_rules()
123	1	notselected_rules = self._get_wanted_not_selected_rules()
124	1	if len(notselected_rules) and not rules:
125	1	raise ValueError(
126		('err- rule(s) "{}" was not selected, '
127		"so there are no results. The rule is"
128		' "notselected" because it'
129		" wasn't a part of the executed profile"
130		" and therefore it wasn't evaluated "
131		"during the scan.")
132		.format(notselected_rules[0]['id_rule']))
133	1	elif not notselected_rules and not rules:
134	1	raise ValueError('err- 404 rule not found!')
135		else:
136	1	return rules
137
138	1	def open_web_browser(self, src):
139	1	if not self.off_webbrowser:
140		src = os.path.join(src, 'index.html')
141		try:
142		webbrowser.get('firefox').open_new_tab(src)
143		except BaseException:
144		webbrowser.open_new_tab(src)
145
146	1	def copy_interpreter(self, dst):
147	1	src = self.get_src('tree_html_interpreter')
148	1	os.mkdir(dst)
149	1	for item in os.listdir(src):
150	1	s = os.path.join(src, item)
151	1	d = os.path.join(dst, item)
152	1	if os.path.isdir(s):
153	1	shutil.copytree(s, d)
154		else:
155	1	shutil.copy2(s, d)
156
157	1	def get_src(self, src):
158	1	_dir = os.path.dirname(os.path.realpath(__file__))
159	1	FIXTURE_DIR = os.path.join(_dir, src)
160	1	return str(FIXTURE_DIR)
161
162	1	def save_dict(self, dict_, src):
163	1	with open(os.path.join(src, 'data.js'), "w+") as data_file:
164	1	data_file.write("var data_json =" + str(json.dumps(
165		dict_, sort_keys=False, indent=4) + ";"))
166
167	1	def get_save_src(self, rule):
168	1	date = str(datetime.now().strftime("-%d_%m_%Y-%H_%M_%S"))
169	1	if self.out is not None:
170	1	if not os.path.isdir(self.out):
171	1	os.mkdir(self.out)
172	1	return os.path.join(self.out, 'graph-of-' + rule + date)
173		return os.path.join(
174		self.out,
175		'graph-of-' + rule + date)
176	1	return os.path.join(
177		os.getcwd(),
178		'graph-of-' + rule + date)
179
180	1	def parse_arguments(self, args):
181	1	self.prepare_parser()
182	1	args = self.parser.parse_args(args)
183	1	return args
184
185	1	def prepare_parser(self):
186	1	self.parser = argparse.ArgumentParser(
187		description=self.MESSAGES.get('description'))
188	1	self.parser.add_argument(
189		'--all',
190		action="store_true",
191		default=False,
192		help="Process all matched rules.")
193	1	self.parser.add_argument(
194		'--hide-passing-tests',
195		action="store_true",
196		default=False,
197		help=(
198		"Do not display passing tests for better orientation in"
199		" graphs that contain a large amount of nodes.(Not implemented)"))
200	1	self.parser.add_argument(
201		'-o',
202		'--output',
203		action="store",
204		default=None,
205		help=self.MESSAGES.get('--output'))
206	1	self.parser.add_argument(
207		"source_filename",
208		help=self.MESSAGES.get('source_filename'))
209	1	self.parser.add_argument(
210		"rule_id", help=(
211		"Rule ID to be visualized. A part from the full rule ID"
212		" a part of the ID or a regular expression can be used."
213		" If brackets are used in the regular expression "
214		"the regular expression must be quoted."))
215

OpenSCAP / oval-graph

Push — master ( be999b...2f9937 )

Client._get_wanted_not_selected_rules() A

Complexity

Size

Duplication

Code Coverage

Importance

Duplication Side-by-Side

Filter issues like