mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 03:14:22 +00:00
9af8f4bb31
This commit is largely based on work by Thiago Arrais (@thiagoarrais) and Shane Bradfield (@l33tLumberjack). I wasn't really able to get their PR reasonably merged onto the latest master, so I reapplied their changes manually here and did some further improvements on that.
17 lines
456 B
Python
17 lines
456 B
Python
"""
|
|
This script rotates all images passing through the proxy by 180 degrees.
|
|
"""
|
|
import io
|
|
|
|
from PIL import Image
|
|
|
|
|
|
def response(flow):
|
|
if flow.response.headers.get("content-type", "").startswith("image"):
|
|
s = io.BytesIO(flow.response.content)
|
|
img = Image.open(s).rotate(180)
|
|
s2 = io.BytesIO()
|
|
img.save(s2, "png")
|
|
flow.response.content = s2.getvalue()
|
|
flow.response.headers["content-type"] = "image/png"
|