hook point for inspecting/modifying dom output prior to finalization

cleandesign-contrib · cleandesign-contrib · commit 8c9704508002 · 2020-06-27T23:36:22.000+10:00
diff --git a/pytest_html/hooks.py b/pytest_html/hooks.py
@@ -2,6 +2,8 @@
 # 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 +23,7 @@ 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. """
diff --git a/pytest_html/plugin.py b/pytest_html/plugin.py
@@ -558,6 +558,8 @@ def generate_summary_item(self):
 
         doc = html.html(head, body)
 
+        session.config.hook.pytest_html_report_final_dom(document=doc)
+
         unicode_doc = "<!DOCTYPE html>\n{}".format(doc.unicode(indent=2))
 
         # Fix encoding issues, e.g. with surrogates
