improve example scripts

This commit is contained in:
Maximilian Hils 2014-08-03 02:39:41 +02:00
parent 3133136da7
commit 8f9395060f

View File

@ -8,10 +8,9 @@ from libmproxy.protocol.http import decoded
def start(ctx, argv): def start(ctx, argv):
if len(argv) != 3: if len(argv) != 3:
sys.exit('Usage: -s "modify-response-body.py old new"') sys.exit('Usage: -s "modify-response-body.py old new"')
global old, new # In larger scripts, a centralized options object (as returned by argparse) is encouraged ctx.old, ctx.new = argv[1], argv[2]
old, new = argv[1:]
def response(ctx, flow): def response(ctx, flow):
with decoded(flow.response): # automatically decode gzipped responses. with decoded(flow.response): # automatically decode gzipped responses.
flow.response.content = flow.response.content.replace(old, new) flow.response.content = flow.response.content.replace(ctx.old, ctx.new)