oval_graph.arf_to_json

Complexity

Total Complexity

10

Size/Duplication

Total Lines	48
Duplicated Lines	0 %

Test Coverage

Coverage

78.38%

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	ArfToJson.run_gui_and_return_answers()	0	11	3
A	ArfToJson.save_dict_as_json()	0	3	2
A	ArfToJson.prepare_data()	0	14	4
A	ArfToJson.create_dict_of_rule()	0	2	1

import webbrowser
import json
import os
import shutil
import pprint
from datetime import datetime
import sys

from .converter import Converter
from .client import Client


class ArfToJson(Client):

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

    def create_dict_of_rule(self, rule_id):
        return self.xml_parser.get_oval_tree(rule_id).save_tree_to_dict()

    def save_dict_as_json(self, dict_, src):
        with open(src + '.json', "w+") as f:
            json.dump(dict_, f)

    def prepare_data(self, rules):
        try:
            out = []
            for rule in rules['rules']:
                oval_tree_dict = self.create_dict_of_rule(rule)
                if self.out is not None:
                    src = self.get_save_src(rule)
                    self.save_dict_as_json(oval_tree_dict, src)
                    out.append(src)
                else:
                    print(str(json.dumps(oval_tree_dict, sort_keys=False, indent=4)))
            return out
        except Exception as error:
            raise ValueError('Rule: "{}" Error: "{}"'.format(rule, error))