diff options
author | Aldo Cortesi <aldo@corte.si> | 2017-04-26 11:01:27 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@corte.si> | 2017-04-26 11:01:27 +1200 |
commit | b72f1390937e9799f588fd9a1564056131fb1aa7 (patch) | |
tree | 2396e9f5c272e671e45e8e4c4abee30ad34b3448 /examples/simple/modify_body_inject_iframe.py | |
parent | f90b4c2ff0f3fd71350900c10dea2a67846e1bdb (diff) | |
download | mitmproxy-b72f1390937e9799f588fd9a1564056131fb1aa7.tar.gz mitmproxy-b72f1390937e9799f588fd9a1564056131fb1aa7.tar.bz2 mitmproxy-b72f1390937e9799f588fd9a1564056131fb1aa7.zip |
configure(options, updated) -> configure(updated)
Options are now available globally on ctx, so the first argument of configure
is redundant.
Diffstat (limited to 'examples/simple/modify_body_inject_iframe.py')
-rw-r--r-- | examples/simple/modify_body_inject_iframe.py | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/examples/simple/modify_body_inject_iframe.py b/examples/simple/modify_body_inject_iframe.py index d54468d2..dff72afa 100644 --- a/examples/simple/modify_body_inject_iframe.py +++ b/examples/simple/modify_body_inject_iframe.py @@ -1,26 +1,21 @@ # (this script works best with --anticache) from bs4 import BeautifulSoup +from mitmproxy import ctx class Injector: - def __init__(self): - self.iframe_url = None - def load(self, loader): loader.add_option( "iframe", str, "", "IFrame to inject" ) - def configure(self, options, updated): - self.iframe_url = options.iframe - def response(self, flow): - if self.iframe_url: + if ctx.options.iframe: html = BeautifulSoup(flow.response.content, "html.parser") if html.body: iframe = html.new_tag( "iframe", - src=self.iframe_url, + src=ctx.options.iframe, frameborder=0, height=0, width=0) |