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

html rewrite: add 'force_html_decl' option, which if set in rewrite_opts, can be used to force an HTML decl, eg. <!DOCTYPE html> if a default one was not provided

Browse Source
This commit is contained in:
Ilya Kreymer 2015-07-18 18:23:57 -07:00
parent 9b08ca9005
commit 0f4d779b00
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pywb/rewrite/html_rewriter.py
View File

@ -109,6 +109,8 @@ class HTMLRewriterMixin(object):
# get opts from urlrewriter # get opts from urlrewriter
self.opts = url_rewriter.rewrite_opts self.opts = url_rewriter.rewrite_opts
self.force_decl = self.opts.get('force_html_decl', None)
self.parsed_any = False self.parsed_any = False
# =========================== # ===========================
@ -300,6 +302,10 @@ class HTMLRewriterMixin(object):
# Clear buffer to create new one for next rewrite() # Clear buffer to create new one for next rewrite()
self.out = None self.out = None
if self.force_decl:
result = self.force_decl + '\n' + result
self.force_decl = None
return result return result
def close(self): def close(self):
@ -409,6 +415,7 @@ class HTMLRewriter(HTMLRewriterMixin, HTMLParser):
def handle_decl(self, data): def handle_decl(self, data):
self.out.write('<!' + data + '>') self.out.write('<!' + data + '>')
self.force_decl = None
def handle_pi(self, data): def handle_pi(self, data):
self.out.write('<?' + data + '>') self.out.write('<?' + data + '>')