From 6a085c2190ee5ec0e60eec6fb7dd4f967d528d0a Mon Sep 17 00:00:00 2001
From: Luna Mendes <luna@elixi.re>
Date: Tue, 20 Nov 2018 17:02:15 -0300
Subject: [PATCH] images: hardcode jpeg/webp

---
 litecord/images.py | 23 ++++++++++++++++++-----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/litecord/images.py b/litecord/images.py
index beaa132..9035578 100644
--- a/litecord/images.py
+++ b/litecord/images.py
@@ -14,17 +14,30 @@ IMAGE_FOLDER = Path('./images')
 log = Logger(__name__)
 
 
-def _get_ext(mime: str):
-    if mime == 'image/webp':
-        return 'webp'
+EXTENSIONS = {
+    'image/jpeg': 'jpeg',
+    'image/webp': 'webp'
+}
+
+
+MIMES = {
+    'jpg': 'image/jpeg',
+    'jpe': 'image/jpeg',
+    'webp': 'image/webp',
+}
+
+
+def _get_ext(mime: str) -> str:
+    if mime in EXTENSIONS:
+        return EXTENSIONS[mime]
 
     extensions = mimetypes.guess_all_extensions(mime)
     return extensions[0].strip('.')
 
 
 def _get_mime(ext: str):
-    if ext == 'webp':
-        return 'image/webp'
+    if ext in MIMES:
+        return MIMES[ext]
 
     return mimetypes.types_map[f'.{ext}']