일부 마인 타입 더 지정

route/main_file.py

@@ -10,11 +10,14 @@ def main_file_2(conn, data):
             data = open('./views/main_css/file/easter_egg.html', 'r').read(),
             menu = 0
         ))
-    elif re.search('\.txt$', data) or data == 'sitemap.xml':
+    elif re.search('\.txt$', data, flags = re.I) or data == 'sitemap.xml':
         if data == 'robots.txt' and not os.path.exists('robots.txt'):
-            return flask.Response('User-agent: *\nDisallow: /\nAllow: /$\nAllow: /w/', mimetype='text/plain')
+            return flask.Response('User-agent: *\nDisallow: /\nAllow: /$\nAllow: /w/', mimetype = 'text/plain')
         elif os.path.exists(data):
-            return flask.send_from_directory('./', data)
+            if re.search('\.txt$', data, flags = re.I):
+                return flask.send_from_directory('./', data, mimetype = 'text/plain')
+            else:
+                return flask.send_from_directory('./', data, mimetype = 'text/xml')
         else:
             return main_error_404.main_error_404_2(conn)
     else:

route/main_image_view.py

@@ -4,6 +4,9 @@ def main_image_view_2(conn, name, app_var):
     curs = conn.cursor()
 
     if os.path.exists(os.path.join(app_var['path_data_image'], name)):
-        return flask.send_from_directory('./' + app_var['path_data_image'], name)
+        return flask.send_from_directory(
+            './' + app_var['path_data_image'], name, 
+            mimetype = 'image/' + re.search('\.([^\.]+)$', name).groups()[0]
+        )
     else:
         return redirect()

route/main_views.py

@@ -18,7 +18,10 @@ def main_views_2(conn, name):
 
     mime_type = re.search('\.([^\.]+)$', rename).groups()[0]
     if mime_type:
-        mime_type = 'text/' + mime_type
+        if mime_type in ['.jpeg', '.jpg', '.gif', '.png', '.webp', '.JPEG', '.JPG', '.GIF', '.PNG', '.WEBP']:
+            mime_type = 'image/' + mime_type
+        else:
+            mime_type = 'text/' + mime_type
     else:
         mime_type = 'text/plain'