| Total Complexity | 5 | 
| Total Lines | 40 | 
| 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("schemas/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>'), | ||
| 29 | encoding='utf-8', | ||
| 30 | standalone=False, | ||
| 31 | with_tail=False, | ||
| 32 | method='html', | ||
| 33 | pretty_print=True) | ||
| 34 | return BytesIO(result) | ||
| 35 | |||
| 36 | 1 | def save_report(self, save_src): | |
| 37 | with open(save_src, 'wb+') as data_file: | ||
| 38 | data_file.writelines(self._get_html()) | ||
| 39 | return [save_src] | ||
| 40 |