From cde65ca8f97024e9b32b1b49624764cf07fa2c49 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <git@voussoir.net>
Date: Tue, 5 Jan 2021 20:41:03 -0800
Subject: [PATCH] Add imagetools.rotate_by_exif.

---
 voussoirkit/imagetools.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/voussoirkit/imagetools.py b/voussoirkit/imagetools.py
index e315bf3..bc53743 100644
--- a/voussoirkit/imagetools.py
+++ b/voussoirkit/imagetools.py
@@ -23,3 +23,38 @@ def fit_into_bounds(
         return (image_width, image_height)
 
     return (new_width, new_height)
+
+def rotate_by_exif(image):
+    '''
+    Rotate the image according to its exif data, so that it will display
+    correctly even if saved without the exif.
+    '''
+    # Thank you Scabbiaza
+    # https://stackoverflow.com/a/26928142
+    import PIL.ExifTags
+
+    for (ORIENTATION_KEY, val) in PIL.ExifTags.TAGS.items():
+        if val == 'Orientation':
+            break
+
+    try:
+        exif = image._getexif()
+    except AttributeError:
+        return image
+
+    if exif is None:
+        return image
+
+    try:
+        rotation = exif[ORIENTATION_KEY]
+    except KeyError:
+        return image
+
+    if rotation == 3:
+        image = image.rotate(180, expand=True)
+    elif rotation == 6:
+        image = image.rotate(270, expand=True)
+    elif rotation == 8:
+        image = image.rotate(90, expand=True)
+
+    return image