backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-15 00:03:28 +01:00
Code Issues Projects Releases Activity

Initialize RegexRewriter with new first_buff

Browse Source
This commit is contained in:
Tessa Walsh 2022-10-04 10:20:05 -04:00
parent dbf52c2579
commit aad10d45fb
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pywb/rewrite/regex_rewriters.py
View File

@ -276,6 +276,9 @@ class JSWombatProxyRewriter(RegexRewriter):
def __init__(self, rewriter, extra_rules=None):
super(JSWombatProxyRewriter, self).__init__(rewriter, extra_rules=extra_rules)
self.rewriter = rewriter
self.extra_rules = extra_rules
self.first_buff = self.rules_factory.first_buff
self.last_buff = self.rules_factory.last_buff
self.local_objs = self.rules_factory.local_objs
@ -303,7 +306,8 @@ class JSWombatProxyRewriter(RegexRewriter):
first_buff = "\nimport {} from '/static/__wb_module_decl.js';\n".format(
", ".join(obj for obj in self.local_objs)
)
return super(JSWombatProxyRewriter, self).rewrite_complete(string, first_buff=first_buff)
super(JSWombatProxyRewriter, self).__init__(self.rewriter, extra_rules=self.extra_rules, first_buff=first_buff)
return super(JSWombatProxyRewriter, self).rewrite_complete(string)
return super(JSWombatProxyRewriter, self).rewrite_complete(string)
# check if any of the wrapped objects are used in the script
# if not, don't rewrite