aboutsummaryrefslogtreecommitdiffstats
path: root/examples/iframe_injector.py
diff options
context:
space:
mode:
authorMaximilian Hils <git@maximilianhils.com>2016-07-16 00:13:58 -0700
committerGitHub <noreply@github.com>2016-07-16 00:13:58 -0700
commitb27d59095d799436fed41eaeaba502ecceb40f76 (patch)
tree152440c1e22850b81aa115817bee4d661f2435de /examples/iframe_injector.py
parent903807292b42b2481a3d72d6dbdc72939fc39b01 (diff)
parente6e39ce80f4daaf6a1d6f8d87616409486d358a5 (diff)
downloadmitmproxy-b27d59095d799436fed41eaeaba502ecceb40f76.tar.gz
mitmproxy-b27d59095d799436fed41eaeaba502ecceb40f76.tar.bz2
mitmproxy-b27d59095d799436fed41eaeaba502ecceb40f76.zip
Merge pull request #1306 from mitmproxy/message-body-encoding
Improve Message Body Encoding
Diffstat (limited to 'examples/iframe_injector.py')
-rw-r--r--examples/iframe_injector.py22
1 files changed, 10 insertions, 12 deletions
diff --git a/examples/iframe_injector.py b/examples/iframe_injector.py
index 70247d31..352c3c24 100644
--- a/examples/iframe_injector.py
+++ b/examples/iframe_injector.py
@@ -2,7 +2,6 @@
# (this script works best with --anticache)
import sys
from bs4 import BeautifulSoup
-from mitmproxy.models import decoded
iframe_url = None
@@ -17,14 +16,13 @@ def start():
def response(flow):
if flow.request.host in iframe_url:
return
- with decoded(flow.response): # Remove content encoding (gzip, ...)
- html = BeautifulSoup(flow.response.content, "lxml")
- if html.body:
- iframe = html.new_tag(
- "iframe",
- src=iframe_url,
- frameborder=0,
- height=0,
- width=0)
- html.body.insert(0, iframe)
- flow.response.content = str(html).encode("utf8")
+ html = BeautifulSoup(flow.response.content, "lxml")
+ if html.body:
+ iframe = html.new_tag(
+ "iframe",
+ src=iframe_url,
+ frameborder=0,
+ height=0,
+ width=0)
+ html.body.insert(0, iframe)
+ flow.response.content = str(html).encode("utf8")