From 36118973d98f1afa81e79a846f595a7c4782ff9a Mon Sep 17 00:00:00 2001 From: harsh vijay Date: Sat, 29 Apr 2017 03:26:14 +0530 Subject: extend mypy example/simple --- examples/simple/modify_body_inject_iframe.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'examples/simple/modify_body_inject_iframe.py') 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: -- cgit v1.2.3