fix: image with non-RGB mode will not be handled correctly

LanceGin · Jul 11, 2019 · a22aa7d · a22aa7d
1 parent 9140236
commit a22aa7d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/haishoku/haillow.py b/haishoku/haillow.py
@@ -14,6 +14,10 @@ def get_image(image_path):
         image_path = io.BytesIO(urllib.request.urlopen(image_path).read())
 
     image = Image.open(image_path)
+
+    # convert non-RGB mode to RGB mode
+    if image.mode != "RGB":
+        image = image.convert("RGB")
     return image
 
 def get_thumbnail(image):