Strip info from Pillow image, correcting rotation

strip_image.py

from PIL import Image


EXIF_ORIENTATION = 274


EXIF_ORIENTATION_TRANSPOSE = [
    None,
    None,
    Image.FLIP_LEFT_RIGHT,
    Image.ROTATE_180,
    Image.FLIP_TOP_BOTTOM,
    Image.TRANSPOSE,
    Image.ROTATE_270,
    Image.TRANSVERSE,
    Image.ROTATE_90,
]


def strip_image(image):
    # TODO: remove color profile? encode to sRGB?
    exif = image.getexif()
    image = Image()._new(image.getdata())
    orientation = exif.get(EXIF_ORIENTATION)
    if orientation:
        transpose = EXIF_ORIENTATION_TRANSPOSE[orientation]
        if transpose is not None:
            image = image.transpose(transpose)
    return image