aboutsummaryrefslogtreecommitdiffstats
path: root/examples/simple/modify_body_inject_iframe.py
diff options
context:
space:
mode:
authorharsh vijay <iharsh234@gmail.com>2017-04-29 03:26:14 +0530
committerGitHub <noreply@github.com>2017-04-29 03:26:14 +0530
commit36118973d98f1afa81e79a846f595a7c4782ff9a (patch)
tree36649bb0b9737548b9b33fd7eac2c359ce1d2d83 /examples/simple/modify_body_inject_iframe.py
parentb537997f4f5d7c33562414a414913b6cd89f99c3 (diff)
downloadmitmproxy-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.py4
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: