From 8e4ab793f9f5723f21ec3cbde9b6e0ea34fcbf11 Mon Sep 17 00:00:00 2001
From: gd <gechandev@gmail.com>
Date: Wed, 5 Jan 2022 07:16:24 +0300
Subject: [PATCH] feat: Refactored uplocad_image(), MIME check skipping added

---
 imgs.py | 38 +++++++++++++++++++++++---------------
 1 file changed, 23 insertions(+), 15 deletions(-)

diff --git a/imgs.py b/imgs.py
index 406f4a8..1b1b9ef 100644
--- a/imgs.py
+++ b/imgs.py
@@ -48,30 +48,38 @@ def upload_image():
     # Handle request from CLI
     if request.files.get('image'):
         file = request.files.get('image')
-        if file.content_type in config['imgs.allowed_mime_types']:
-            image_name = upload_file(file)
-            return get_image_url(image_name) + '\n'
-        else:
-            # Prevent recource leek. Force close buffered file
-            request.body.close()
-            response.status = 415
-            return 'Error: bad file MIME type\n'
+        rq = 'cli'
     # Handle request from web-browser
     elif request.files.get('image_web'):
         file = request.files.get('image_web')
+        rq = 'web'
+
+    if config['imgs.allowed_mime_types'] == '*':
+        # Skip MIME checking.
+        image_name = upload_file(file)
+    else:
         if file.content_type in config['imgs.allowed_mime_types']:
+            # Upload file!
             image_name = upload_file(file)
-            return template('index.tpl',
-                uploaded = True, not_found = False, bad_mime_type = False,
-                base_url = get_base_url(), image_url = get_image_url(image_name))
         else:
+            # Show MIME type error!
             # Prevent recource leek. Force close buffered file
             request.body.close()
             response.status = 415
-            return template('index.tpl',
-                uploaded = False, not_found = False, bad_mime_type = True,
-                allowed_mime_types = config['imgs.allowed_mime_types'],
-                base_url = get_base_url(), image_url = 'None')
+            if rq == 'cli':
+                return 'Error: bad file MIME type\n'
+            else:
+                return template('index.tpl',
+                    uploaded = False, not_found = False, bad_mime_type = True,
+                    allowed_mime_types = config['imgs.allowed_mime_types'],
+                    base_url = get_base_url(), image_url = 'None')
+    # Return 200 OK
+    if rq == 'cli':
+        return get_image_url(image_name) + '\n'
+    else:
+        return template('index.tpl',
+            uploaded = True, not_found = False, bad_mime_type = False,
+            base_url = get_base_url(), image_url = get_image_url(image_name))
 
 @get('/<image_name>')
 def send_image(image_name):