Просмотр исходного кода

https://github.com/openNAMU/openNAMU/issues/1747

잉여개발기 (SPDV) 3 лет назад
Родитель
Сommit
6d2adc3134
3 измененных файлов с 39 добавлено и 37 удалено
  1. 26 25
      route/main_view.py
  2. 12 11
      route/main_view_image.py
  3. 1 1
      version.json

+ 26 - 25
route/main_view.py

@@ -2,32 +2,33 @@ from .tool.func import *
 from .main_func_error_404 import main_func_error_404
 
 def main_view(name = ''):
-    file_name = re.search(r'([^/]+)$', name)
-    if not file_name:
-        return main_func_error_404()
-    else:
-        file_name = file_name.group(1)
-        dir_name = './views/' + re.sub(r'\.{2,}', '', re.sub(r'([^/]+)$', '', name))
-
-        mime_type = file_name.split('.')
-        if len(mime_type) < 2:
-            mime_type = 'text/plain'
+    with get_db_connect() as conn:
+        file_name = re.search(r'([^/]+)$', name)
+        if not file_name:
+            return main_func_error_404()
         else:
-            mime_type = mime_type[len(mime_type) - 1].lower()
-            image_type = ['jpeg', 'jpg', 'gif', 'png', 'webp', 'ico', 'svg']
-            if mime_type in image_type:
-                if not mime_type == 'svg':
-                    mime_type = 'image/' + mime_type
-                else:
-                    mime_type = 'image/svg+xml'
-            elif mime_type == 'js':
-                mime_type = 'text/javascript'
-            elif mime_type == 'txt':
+            file_name = file_name.group(1)
+            dir_name = './views/' + re.sub(r'\.{2,}', '', re.sub(r'([^/]+)$', '', name))
+
+            mime_type = file_name.split('.')
+            if len(mime_type) < 2:
                 mime_type = 'text/plain'
             else:
-                mime_type = 'text/' + mime_type
+                mime_type = mime_type[len(mime_type) - 1].lower()
+                image_type = ['jpeg', 'jpg', 'gif', 'png', 'webp', 'ico', 'svg']
+                if mime_type in image_type:
+                    if not mime_type == 'svg':
+                        mime_type = 'image/' + mime_type
+                    else:
+                        mime_type = 'image/svg+xml'
+                elif mime_type == 'js':
+                    mime_type = 'text/javascript'
+                elif mime_type == 'txt':
+                    mime_type = 'text/plain'
+                else:
+                    mime_type = 'text/' + mime_type
 
-        return flask.send_from_directory(
-            dir_name, file_name, 
-            mimetype = mime_type
-        )
+            return flask.send_from_directory(
+                dir_name, file_name, 
+                mimetype = mime_type
+            )

+ 12 - 11
route/main_view_image.py

@@ -2,15 +2,16 @@ from .tool.func import *
 from .main_func_error_404 import main_func_error_404
 
 def main_view_image(name = ''):
-    mime_type = re.search(r'([^.]+)$', name)
-    if mime_type:
-        mime_type = mime_type.group(1).lower()
-        if mime_type == 'svg':
-            mime_type = 'svg+xml'
+    with get_db_connect() as conn:
+        mime_type = re.search(r'([^.]+)$', name)
+        if mime_type:
+            mime_type = mime_type.group(1).lower()
+            if mime_type == 'svg':
+                mime_type = 'svg+xml'
 
-        return flask.send_from_directory(
-            './' + load_image_url(), name, 
-            mimetype = 'image/' + mime_type
-        )
-    else:
-        return main_func_error_404()
+            return flask.send_from_directory(
+                './' + load_image_url(), name, 
+                mimetype = 'image/' + mime_type
+            )
+        else:
+            return main_func_error_404()

+ 1 - 1
version.json

@@ -1,6 +1,6 @@
 {
     "beta" : {
-        "r_ver" : "v3.4.6-RC3-dev154",
+        "r_ver" : "v3.4.6-RC3-dev155",
         "c_ver" : "3500361",
         "s_ver" : "3500111"
     }