diff --git a/pywb/rewrite/html_rewriter.py b/pywb/rewrite/html_rewriter.py index afbeba6c..527b0f07 100644 --- a/pywb/rewrite/html_rewriter.py +++ b/pywb/rewrite/html_rewriter.py @@ -109,6 +109,8 @@ class HTMLRewriterMixin(object): # get opts from urlrewriter self.opts = url_rewriter.rewrite_opts + self.force_decl = self.opts.get('force_html_decl', None) + self.parsed_any = False # =========================== @@ -300,6 +302,10 @@ class HTMLRewriterMixin(object): # Clear buffer to create new one for next rewrite() self.out = None + if self.force_decl: + result = self.force_decl + '\n' + result + self.force_decl = None + return result def close(self): @@ -409,6 +415,7 @@ class HTMLRewriter(HTMLRewriterMixin, HTMLParser): def handle_decl(self, data): self.out.write('') + self.force_decl = None def handle_pi(self, data): self.out.write('')