diff options
Diffstat (limited to 'examples/iframe_injector.py')
| -rw-r--r-- | examples/iframe_injector.py | 24 |
1 files changed, 11 insertions, 13 deletions
diff --git a/examples/iframe_injector.py b/examples/iframe_injector.py index 9495da93..5803b4c1 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 def start(context): @@ -14,15 +13,14 @@ def start(context): def response(context, flow): if flow.request.host in context.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=context.iframe_url, - frameborder=0, - height=0, - width=0) - html.body.insert(0, iframe) - flow.response.content = str(html) - context.log("Iframe inserted.") + html = BeautifulSoup(flow.response.content, "lxml") + if html.body: + iframe = html.new_tag( + "iframe", + src=context.iframe_url, + frameborder=0, + height=0, + width=0) + html.body.insert(0, iframe) + flow.response.content = str(html) + context.log("Iframe inserted.") |