oval_graph.command_line

Complexity

Total Complexity

9

Size/Duplication

Total Lines	46
Duplicated Lines	0 %

Test Coverage

Coverage

0%

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	print_where_is_saved_result()	0	4	2
A	arf_to_graph()	0	5	2
A	main()	0	10	3
A	arf_to_json()	0	5	2

import sys

from .arf_to_html import ArfToHtml
from .arf_to_json import ArfToJson


def print_where_is_saved_result(results_src):
    print("Results are saved:")
    for src in results_src:
        print(src)


def arf_to_graph(args=None):
    if args is not None:
        main(ArfToHtml(args))
    else:
        main(ArfToHtml(sys.argv[1:]))


def arf_to_json(args=None):
    if args is not None:
        main(ArfToJson(args))
    else:
        main(ArfToJson(sys.argv[1:]))


def main(client):
    rules = client.search_rules_id()
    if len(rules) > 1:
        answers = client.run_gui_and_return_answers()
        if answers is not None:
            results_src = client.prepare_data(answers)
            print_where_is_saved_result(results_src)
    else:
        results_src = client.prepare_data({'rules': [rules[0]['id_rule']]})
        print_where_is_saved_result(results_src)


if __name__ == '__main__':
    if sys.argv[1] == "arf-to-graph":
        arf_to_graph(sys.argv[2:])
    if sys.argv[1] == "arf-to-json":
        arf_to_json(sys.argv[2:])
    else:
        print("err- Bad command!")