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


state = {}


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.
    state["old"], state["new"] = sys.argv[1].encode(), sys.argv[2].encode()


def response(flow):
    flow.response.content = flow.response.content.replace(
        state["old"],
        state["new"]
    )