diff options
Diffstat (limited to 'examples/modify_response_body.py')
-rw-r--r-- | examples/modify_response_body.py | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/examples/modify_response_body.py b/examples/modify_response_body.py index 3034892e..b4632248 100644 --- a/examples/modify_response_body.py +++ b/examples/modify_response_body.py @@ -2,19 +2,20 @@ # (this script works best with --anticache) import sys -from mitmproxy.models import decoded +state = {} -def start(context): + +def start(): if len(sys.argv) != 3: raise ValueError('Usage: -s "modify_response_body.py old new"') # You may want to use Python's argparse for more sophisticated argument # parsing. - context.old, context.new = sys.argv[1], sys.argv[2] + state["old"], state["new"] = sys.argv[1].encode(), sys.argv[2].encode() -def response(context, flow): - with decoded(flow.response): # automatically decode gzipped responses. - flow.response.content = flow.response.content.replace( - context.old, - context.new) +def response(flow): + flow.response.content = flow.response.content.replace( + state["old"], + state["new"] + ) |