2016-03-20 18:40:03 +00:00
|
|
|
from six.moves import cStringIO as StringIO
|
2014-01-04 02:15:08 +00:00
|
|
|
from PIL import Image
|
|
|
|
|
|
2015-05-30 00:03:28 +00:00
|
|
|
|
2016-07-08 01:37:33 +00:00
|
|
|
def response(flow):
|
2015-09-05 18:45:58 +00:00
|
|
|
if flow.response.headers.get("content-type", "").startswith("image"):
|
2016-07-02 09:01:46 +00:00
|
|
|
try:
|
|
|
|
|
s = StringIO(flow.response.content)
|
|
|
|
|
img = Image.open(s).rotate(180)
|
|
|
|
|
s2 = StringIO()
|
|
|
|
|
img.save(s2, "png")
|
|
|
|
|
flow.response.content = s2.getvalue()
|
|
|
|
|
flow.response.headers["content-type"] = "image/png"
|
|
|
|
|
except: # Unknown image types etc.
|
|
|
|
|
pass
|
