aboutsummaryrefslogtreecommitdiffstats
path: root/examples/modify_response_body.py
blob: d68bcf63eb739d889cb434bb99e65a962d5d370a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# Usage: mitmdump -s "modify_response_body.py mitmproxy bananas"
# (this script works best with --anticache)
from mitmproxy.models import decoded


def start(context, argv):
    if len(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 = argv[1], argv[2]


def response(context, flow):
    with decoded(flow.response):  # automatically decode gzipped responses.
        flow.response.content = flow.response.content.replace(
            context.old,
            context.new)