Total Complexity | 5 |
Total Lines | 41 |
Duplicated Lines | 0 % |
Coverage | 34.78% |
Changes | 0 |
1 | 1 | from io import BytesIO |
|
|
|||
2 | |||
3 | 1 | import lxml.html |
|
4 | 1 | from lxml import etree |
|
5 | |||
6 | |||
7 | 1 | class BuilderHtmlReport(): |
|
8 | |||
9 | 1 | def __init__(self, display_html, xml_parser, arf_file_src): |
|
10 | self.display_html = display_html |
||
11 | self.arf_file_src = arf_file_src |
||
12 | self.src_xsl = xml_parser.get_src("xsl/xccdf-report.xsl") |
||
13 | self.xslt_doc = etree.parse(self.src_xsl) |
||
14 | self.xslt_transformer = etree.XSLT(self.xslt_doc) |
||
15 | |||
16 | self.source_doc = etree.parse(self.arf_file_src) |
||
17 | self.output_doc = self.xslt_transformer(self.source_doc) |
||
18 | |||
19 | 1 | def _get_html_object(self): |
|
20 | html_object = lxml.html.fromstring(str(self.output_doc)) |
||
21 | return html_object |
||
22 | |||
23 | 1 | def _get_html(self): |
|
24 | html_object = self._get_html_object() |
||
25 | result = etree.tostring( |
||
26 | html_object, |
||
27 | xml_declaration=True, |
||
28 | doctype=('<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN"' |
||
29 | ' "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'), |
||
30 | encoding='utf-8', |
||
31 | standalone=False, |
||
32 | with_tail=False, |
||
33 | method='html', |
||
34 | pretty_print=True) |
||
35 | return BytesIO(result) |
||
36 | |||
37 | 1 | def save_report(self, save_src): |
|
38 | with open(save_src, 'wb+') as data_file: |
||
39 | data_file.writelines(self._get_html()) |
||
40 | return [save_src] |
||
41 |