tz/ERP
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复上传头像时格式错误的问题

Browse Source
This commit is contained in:
zzh 2025-11-21 21:44:28 +08:00
parent c9efcd3571
commit 402a83e651
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server/app.py
View File

@ -387,11 +387,13 @@ def upload_avatar():
# 验证文件类型 # 验证文件类型
allowed_extensions = {'png', 'jpg', 'jpeg', 'gif', 'webp'} allowed_extensions = {'png', 'jpg', 'jpeg', 'gif', 'webp'}
filename = secure_filename(file.filename)
if '.' not in filename: # 先从原始文件名获取扩展名
original_filename = file.filename
if '.' not in original_filename:
return jsonify({'error': '无效的文件格式'}), 400 return jsonify({'error': '无效的文件格式'}), 400
ext = filename.rsplit('.', 1)[1].lower() ext = original_filename.rsplit('.', 1)[1].lower()
if ext not in allowed_extensions: if ext not in allowed_extensions:
return jsonify({'error': '不支持的文件格式，请上传 PNG、JPG、GIF 或 WEBP 格式'}), 400 return jsonify({'error': '不支持的文件格式，请上传 PNG、JPG、GIF 或 WEBP 格式'}), 400