voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Let rotate_by_exif return new exif with orientation=1.

Browse source
This commit is contained in:
voussoir 2021-06-06 20:29:29 -07:00
parent 663e65795a
commit 689b8f5718
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 29 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
voussoirkit/imagetools.py
View file

@ -1,3 +1,4 @@
import copy
import PIL.ExifTags import PIL.ExifTags
ORIENTATION_KEY = None ORIENTATION_KEY = None
@ -35,6 +36,9 @@ def rotate_by_exif(image):
''' '''
Rotate the image according to its exif data, so that it will display Rotate the image according to its exif data, so that it will display
correctly even if saved without the exif. correctly even if saved without the exif.
Returns (image, exif) where exif has the orientation key set to 1,
the upright position, if the rotation was successful.
''' '''
# Thank you Scabbiaza # Thank you Scabbiaza
# https://stackoverflow.com/a/26928142 # https://stackoverflow.com/a/26928142
@ -42,21 +46,38 @@ def rotate_by_exif(image):
try: try:
exif = image.getexif() exif = image.getexif()
except AttributeError: except AttributeError:
return image return (image, exif)
if exif is None: if exif is None:
return image return (image, exif)
try: try:
rotation = exif[ORIENTATION_KEY] rotation = exif[ORIENTATION_KEY]
except KeyError: except KeyError:
return image return (image, exif)
if rotation == 3: exif = copy.deepcopy(exif)
image = image.rotate(180, expand=True)
if rotation == 1:
pass
elif rotation == 2:
image = image.transpose(PIL.Image.FLIP_LEFT_RIGHT)
elif rotation == 3:
image = image.transpose(PIL.Image.ROTATE_180)
elif rotation == 4:
image = image.transpose(PIL.Image.FLIP_LEFT_RIGHT)
image = image.transpose(PIL.Image.ROTATE_180)
elif rotation == 5:
image = image.transpose(PIL.Image.FLIP_LEFT_RIGHT)
image = image.transpose(PIL.Image.ROTATE_90)
elif rotation == 6: elif rotation == 6:
image = image.rotate(270, expand=True) image = image.transpose(PIL.Image.ROTATE_270)
elif rotation == 7:
image = image.transpose(PIL.Image.FLIP_LEFT_RIGHT)
image = image.transpose(PIL.Image.ROTATE_270)
elif rotation == 8: elif rotation == 8:
image = image.rotate(90, expand=True) image = image.transpose(PIL.Image.ROTATE_90)
return image exif[ORIENTATION_KEY] = 1
return (image, exif)