diff options
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: |