diff options
Diffstat (limited to 'examples')
-rw-r--r-- | examples/simple/custom_option.py | 4 | ||||
-rw-r--r-- | examples/simple/modify_body_inject_iframe.py | 11 |
2 files changed, 5 insertions, 10 deletions
diff --git a/examples/simple/custom_option.py b/examples/simple/custom_option.py index c8bc98d4..5b6070dd 100644 --- a/examples/simple/custom_option.py +++ b/examples/simple/custom_option.py @@ -6,6 +6,6 @@ def load(l): l.add_option("custom", bool, False, "A custom option") -def configure(options, updated): +def configure(updated): if "custom" in updated: - ctx.log.info("custom option value: %s" % options.custom) + ctx.log.info("custom option value: %s" % ctx.options.custom) 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) |