修复base64图片上传接口的图片格式验证问题

2021-10-24 23:32:36 +08:00 · 2021-10-24 23:32:36 +08:00 · dcb4fcb23d
commit dcb4fcb23d
parent 46ef5326e4
1 changed files with 2 additions and 2 deletions
--- a/app_doc/util_upload_img.py
+++ b/app_doc/util_upload_img.py
@ -228,8 +228,8 @@ def getImageExtensionName(temps):
        #image/png
        temps = temps[0].split('image/')
        if len(temps) == 2:
-            ## 如果文件传了扩展名，就取扩展名的文件类型
-            if  temps[-1] != '':
+            ## 如果文件传了扩展名，就取扩展名的文件类型，判断图片格式是否允许上传
+            if temps[-1] in settings.ALLOWED_IMG:
                return "." + temps[-1]
    return ".png"