blob: b7489cb656874e6202e77d0b6ab99585efecceae (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
import cStringIO
from PIL import Image
from mitmproxy.models import decoded
def response(context, flow):
if flow.response.headers.get("content-type", "").startswith("image"):
with decoded(flow.response): # automatically decode gzipped responses.
try:
s = cStringIO.StringIO(flow.response.content)
img = Image.open(s).rotate(180)
s2 = cStringIO.StringIO()
img.save(s2, "png")
flow.response.content = s2.getvalue()
flow.response.headers["content-type"] = "image/png"
except: # Unknown image types etc.
pass
|
