WIP: Add custom_banner to head_insert

webrecorder · Oct 10, 2022 · f92fe76 · f92fe76
1 parent a251082
commit f92fe76
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/pywb/templates/custom_banner.html b/pywb/templates/custom_banner.html
@@ -0,0 +1 @@
+<!-- Add custom banner here. Used only in non-framed replay. -->
diff --git a/pywb/templates/head_insert.html b/pywb/templates/head_insert.html
@@ -62,6 +62,12 @@
 
 {% endif %}
 
+{% if not wbinfo.is_framed %}
+
+{{ custom_banner_html }}
+
+{% endif %}
+
 {% endautoescape %}
 
 <!-- End WB Insert -->

diff --git a/tests/test_auto_colls.py b/tests/test_auto_colls.py
@@ -279,7 +279,7 @@ def test_add_custom_banner(self):
         """
 
         banner_file = os.path.join(self.root_dir, COLLECTIONS, 'test',
-                                   'templates', 'banner.html')
+                                   'templates', 'custom_banner.html')
 
         with open(banner_file, 'w+b') as fh:
             fh.write(b'<div>Custom Banner Here!</div>')