Mitmproxy image download script

save_images.py

from mitmproxy import http
import os
import hashlib


def response(flow: http.HTTPFlow) -> None:
    save_dir = "downloaded_images"
    os.makedirs(save_dir, exist_ok=True)

    content_type = flow.response.headers.get("Content-Type", "")
    if "image" in content_type:
        url_hash = hashlib.sha256(flow.request.url.encode('utf-8')).hexdigest()
        file_extension = content_type.split('/')[-1]
        filename = f"{url_hash}.{file_extension}"
        file_path = os.path.join(save_dir, filename)

        with open(file_path, "wb") as f:
            f.write(flow.response.content)
        print(f"Image saved: {file_path}")