diff --git a/pywb/webapp/handlers.py b/pywb/webapp/handlers.py index 2e074d66..ce11b6b1 100644 --- a/pywb/webapp/handlers.py +++ b/pywb/webapp/handlers.py @@ -14,7 +14,7 @@ from pywb.framework.wbrequestresponse import WbResponse #================================================================= class WBHandler(WbUrlHandler): def __init__(self, index_reader, replay, - search_view=None): + search_view=None, config=None): self.index_reader = index_reader @@ -40,9 +40,10 @@ class WBHandler(WbUrlHandler): cdx_lines, cdx_callback) - def render_search_page(self, wbrequest): + def render_search_page(self, wbrequest, **kwargs): if self.search_view: - return self.search_view.render_response(wbrequest=wbrequest) + return self.search_view.render_response(wbrequest=wbrequest, + **kwargs) else: return WbResponse.text_response('No Lookup Url Specified') diff --git a/pywb/webapp/pywb_init.py b/pywb/webapp/pywb_init.py index bc942966..544433bd 100644 --- a/pywb/webapp/pywb_init.py +++ b/pywb/webapp/pywb_init.py @@ -85,10 +85,13 @@ def create_wb_handler(query_handler, config, create_template(config.get('head_insert_html'), 'Head Insert')) + defmod = config.get('default_mod', '') + replayer = ReplayView( content_loader=resolving_loader, - content_rewriter=RewriteContent(ds_rules_file=ds_rules_file), + content_rewriter=RewriteContent(ds_rules_file=ds_rules_file, + defmod=defmod), head_insert_view=head_insert_view, @@ -111,6 +114,7 @@ def create_wb_handler(query_handler, config, query_handler, replayer, search_view=search_view, + config=config, ) return wb_handler