# ! IMPORT ! make sure you ran mitmproxy with this script, 
# eg: `/path/to/mitmproxy -s mitmproxy.py`

import uuid
from mitmproxy import http

from config.helper import config

class Writer:
    def response(self, flow: http.HTTPFlow) -> None:
        if flow.request.host == 'live.douyin.com':
            with open(config().mitmproxy.log_dir + uuid.uuid4().hex, 'wb') as f:
                f.write(bytes(flow.response.content))

addons = [Writer()]