import sys
from mitmproxy import http
import urllib.parse
from mitmproxy.tools.main import mitmdump

class Addon:
    def request(self, flow: http.HTTPFlow):
        # 匹配提交url
        if flow.request.pretty_url.startswith("http://baodao.*.edu.cn/XSSC/Question.aspx?tid="):
            # 拦截请求，修改body
            old_body = flow.request.get_text()

            # 解析原来的POST数据，它是一个字符串，我们需要将它转换为一个字典
            data = urllib.parse.parse_qs(old_body)

            # 修改判断答案是否正确参数
            key = 'ctl00$ContentPlaceHolder1$RBL_Answer'
            if key in data:
                data[key] = ['1']  # dictionary的值需要是一个列表

            # 把修改后的字典再转换回来为字符串
            new_body = urllib.parse.urlencode(data, doseq=True)

            # 更新请求体
            flow.request.set_text(new_body)

addons = [
    Addon()
]

if __name__ == "__main__":
    sys.argv.append('-s')
    sys.argv.append(__file__)
    mitmdump()