From 633eb31f57bbe992616fe45a5d012e97d4365c3c Mon Sep 17 00:00:00 2001 From: Jack Cushman Date: Thu, 22 Oct 2015 16:40:59 -0400 Subject: [PATCH] Use webencodings to encode head_insert_str. --- pywb/rewrite/rewrite_content.py | 3 ++- setup.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pywb/rewrite/rewrite_content.py b/pywb/rewrite/rewrite_content.py index 08f8a208..171a7c0c 100644 --- a/pywb/rewrite/rewrite_content.py +++ b/pywb/rewrite/rewrite_content.py @@ -1,5 +1,6 @@ #import chardet import pkgutil +import webencodings import yaml import re @@ -164,7 +165,7 @@ class RewriteContent: if charset: try: - head_insert_str = head_insert_orig.encode(charset) + head_insert_str = webencodings.encode(head_insert_orig, charset) except: pass diff --git a/setup.py b/setup.py index 3940b632..959d5852 100755 --- a/setup.py +++ b/setup.py @@ -76,7 +76,8 @@ setup( 'jinja2', 'surt', 'pyyaml', - 'watchdog' + 'watchdog', + 'webencodings', ], tests_require=[ 'pytest',