hook point for inspecting/modifying dom output prior to finalization

cleandesign-contrib · cleandesign-contrib · commit b71bbf9030d1 · 2020-06-27T23:06:52.000+10:00
diff --git a/pytest_html/hooks.py b/pytest_html/hooks.py
@@ -2,6 +2,7 @@
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
+from py.xml import html
 
 def pytest_html_report_title(report):
     """ Called before adding the title to the report """
@@ -21,3 +22,6 @@ def pytest_html_results_table_row(report, cells):
 
 def pytest_html_results_table_html(report, data):
     """ Called after building results table additional HTML. """
+
+def pytest_html_report_final_dom(document : html.html):
+    """ Called after the entire document has been created. Allows inspection and modification of DOM. """
diff --git a/pytest_html/plugin.py b/pytest_html/plugin.py
@@ -554,6 +554,8 @@ def generate_summary_item(self):
 
         doc = html.html(head, body)
 
+        session.config.hook.pytest_html_report_complete(document=doc)
+
         unicode_doc = "<!DOCTYPE html>\n{}".format(doc.unicode(indent=2))
 
         # Fix encoding issues, e.g. with surrogates
