diff options
Diffstat (limited to 'examples/iframe_injector.py')
| -rw-r--r-- | examples/iframe_injector.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/examples/iframe_injector.py b/examples/iframe_injector.py index ad844f19..9495da93 100644 --- a/examples/iframe_injector.py +++ b/examples/iframe_injector.py @@ -1,13 +1,14 @@ # Usage: mitmdump -s "iframe_injector.py url" # (this script works best with --anticache) +import sys from bs4 import BeautifulSoup from mitmproxy.models import decoded -def start(context, argv): - if len(argv) != 2: +def start(context): + if len(sys.argv) != 2: raise ValueError('Usage: -s "iframe_injector.py url"') - context.iframe_url = argv[1] + context.iframe_url = sys.argv[1] def response(context, flow): |