| Total Complexity | 5 |
| Total Lines | 34 |
| Duplicated Lines | 0 % |
| Coverage | 100% |
| Changes | 0 | ||
| 1 | 1 | from datetime import datetime |
|
| 2 | |||
| 3 | |||
| 4 | 1 | from .client import Client |
|
| 5 | |||
| 6 | |||
| 7 | 1 | class ArfToHtml(Client): |
|
| 8 | 1 | def __init__(self, args): |
|
| 9 | 1 | super().__init__(args) |
|
| 10 | 1 | self.off_webbrowser = self.arg.off_web_browser |
|
| 11 | |||
| 12 | 1 | def prepare_data(self, rules): |
|
| 13 | 1 | try: |
|
| 14 | 1 | out = [] |
|
| 15 | 1 | for rule in rules['rules']: |
|
| 16 | 1 | oval_tree_dict = self.create_dict_of_rule(rule) |
|
| 17 | 1 | src = self.get_save_src(rule) |
|
| 18 | 1 | self.copy_interpreter(src) |
|
| 19 | 1 | self.save_dict(oval_tree_dict, src) |
|
| 20 | 1 | self.open_web_browser(src) |
|
| 21 | 1 | print('Rule "{}" done!'.format(rule)) |
|
| 22 | 1 | out.append(src) |
|
| 23 | 1 | return out |
|
| 24 | 1 | except Exception as error: |
|
| 25 | 1 | raise ValueError('Rule: "{}" Error: "{}"'.format(rule, error)) |
|
| 26 | |||
| 27 | 1 | def prepare_parser(self): |
|
| 28 | 1 | super().prepare_parser() |
|
| 29 | 1 | self.parser.add_argument( |
|
| 30 | '--off-web-browser', |
||
| 31 | action="store_true", |
||
| 32 | default=False, |
||
| 33 | help="It does not start the web browser.") |
||
| 34 |