| 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 |