Total Complexity | 13 |
Total Lines | 61 |
Duplicated Lines | 0 % |
Coverage | 0% |
Changes | 0 |
1 | import sys |
||
2 | |||
3 | from .arf_to_html import ArfToHtml |
||
4 | from .arf_to_json import ArfToJson |
||
5 | from .json_to_html import JsonToHtml |
||
6 | |||
7 | |||
8 | def print_where_is_saved_result(results_src): |
||
9 | if results_src: |
||
10 | print("Results are saved:") |
||
11 | for src in results_src: |
||
12 | print(src) |
||
13 | |||
14 | |||
15 | def arf_to_graph(args=None): |
||
16 | if args is not None: |
||
17 | main(ArfToHtml(args)) |
||
18 | else: |
||
19 | main(ArfToHtml(sys.argv[1:])) |
||
20 | |||
21 | |||
22 | def arf_to_json(args=None): |
||
23 | if args is not None: |
||
24 | main(ArfToJson(args)) |
||
25 | else: |
||
26 | main(ArfToJson(sys.argv[1:])) |
||
27 | |||
28 | |||
29 | def json_to_graph(args=None): |
||
30 | if args is not None: |
||
31 | main(JsonToHtml(args)) |
||
32 | else: |
||
33 | main(JsonToHtml(sys.argv[1:])) |
||
34 | |||
35 | |||
36 | def main(client): |
||
37 | rules = [] |
||
38 | results_src = [] |
||
39 | if not isinstance(client, JsonToHtml): |
||
40 | rules = client.search_rules_id() |
||
41 | if len(rules) > 1: |
||
42 | answers = client.run_gui_and_return_answers() |
||
43 | if answers is not None: |
||
44 | results_src = client.prepare_data(answers) |
||
45 | else: |
||
46 | results_src = client.prepare_data({'rules': [rules[0]['id_rule']]}) |
||
47 | else: |
||
48 | results_src = client.prepare_data() |
||
49 | print_where_is_saved_result(results_src) |
||
50 | |||
51 | |||
52 | if __name__ == '__main__': |
||
53 | if sys.argv[1] == "arf-to-graph": |
||
54 | arf_to_graph(sys.argv[2:]) |
||
55 | elif sys.argv[1] == "arf-to-json": |
||
56 | arf_to_json(sys.argv[2:]) |
||
57 | elif sys.argv[1] == "json-to-graph": |
||
58 | json_to_graph(sys.argv[2:]) |
||
59 | else: |
||
60 | print("err- Bad command!") |
||
61 |