diff options
author | harsh vijay <iharsh234@gmail.com> | 2017-04-29 03:26:14 +0530 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-04-29 03:26:14 +0530 |
commit | 36118973d98f1afa81e79a846f595a7c4782ff9a (patch) | |
tree | 36649bb0b9737548b9b33fd7eac2c359ce1d2d83 /examples/simple/modify_body_inject_iframe.py | |
parent | b537997f4f5d7c33562414a414913b6cd89f99c3 (diff) | |
download | mitmproxy-36118973d98f1afa81e79a846f595a7c4782ff9a.tar.gz mitmproxy-36118973d98f1afa81e79a846f595a7c4782ff9a.tar.bz2 mitmproxy-36118973d98f1afa81e79a846f595a7c4782ff9a.zip |
extend mypy example/simple
Diffstat (limited to 'examples/simple/modify_body_inject_iframe.py')
-rw-r--r-- | examples/simple/modify_body_inject_iframe.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/simple/modify_body_inject_iframe.py b/examples/simple/modify_body_inject_iframe.py index dff72afa..595bd9f2 100644 --- a/examples/simple/modify_body_inject_iframe.py +++ b/examples/simple/modify_body_inject_iframe.py @@ -1,6 +1,6 @@ # (this script works best with --anticache) from bs4 import BeautifulSoup -from mitmproxy import ctx +from mitmproxy import ctx, http class Injector: @@ -9,7 +9,7 @@ class Injector: "iframe", str, "", "IFrame to inject" ) - def response(self, flow): + def response(self, flow: http.HTTPFlow) -> None: if ctx.options.iframe: html = BeautifulSoup(flow.response.content, "html.parser") if html.body: |