修复上传图片校验失败的问题

This commit is contained in:
梦凌汐 2025-06-05 02:12:26 +08:00
parent 7d79e4a3c9
commit 7c6eed2022
3 changed files with 22 additions and 22 deletions

View File

@ -26,7 +26,7 @@ CatismImage 是一个基于Web的图片存储和管理系统提供用户友
4. [可选] 配置NSFWPY 4. [可选] 配置NSFWPY
如果您希望启用NSFW过滤功能您需要配置NSFWPY。 如果您希望启用NSFW过滤功能您需要配置NSFWPY。
请参考[HG-ha/nsfwpy](https://github.com/HG-ha/nsfwpy),通过您喜欢的方式完成配置。 请参考[HG-ha/nsfwpy](https://github.com/HG-ha/nsfwpy),通过您喜欢的方式完成配置。
5. 配置环境变量: 5. 配置环境变量:

View File

@ -60,30 +60,30 @@ def upload_image():
file.seek(0) file.seek(0)
except Exception: except Exception:
return jsonify({'error': '无效的图片文件'}), 400 return jsonify({'error': '无效的图片文件'}), 400
# 如果启用了NSFW检查
if AppConfig.ENABLE_NSFW_FILTER:
print("NSFW检查已启用")
file.seek(0)
try:
response = requests.post(
f"{AppConfig.NSFWPY_ENDPOINT}/classify",
files={'image': file},
headers={'accept': 'application/json'}
)
response.raise_for_status()
nsfw_data = response.json()
# 检查NSFW评分
if nsfw_data.get('neutral', 0) + nsfw_data.get('drawing', 0) < 0.5:
return jsonify({'error': '图片内容不符合安全标准'}), 403
file.seek(0)
except Exception as e:
return jsonify({'error': f'NSFW检查失败: {str(e)}'}), 500
except Exception: except Exception:
return jsonify({'error': '无效的图片文件'}), 400 return jsonify({'error': '无效的图片文件'}), 400
# 如果启用了NSFW检查
if AppConfig.ENABLE_NSFW_FILTER:
print("NSFW检查已启用")
file.seek(0)
try:
response = requests.post(
f"{AppConfig.NSFWPY_ENDPOINT}/classify",
files={'image': file},
headers={'accept': 'application/json'}
)
response.raise_for_status()
nsfw_data = response.json()
# 检查NSFW评分
if nsfw_data.get('neutral', 0) + nsfw_data.get('drawing', 0) < 0.5:
return jsonify({'error': '图片内容不符合安全标准'}), 403
file.seek(0)
except Exception as e:
return jsonify({'error': f'NSFW检查失败: {str(e)}'}), 500
# 获取文件大小并检查存储空间 # 获取文件大小并检查存储空间
file.seek(0, 2) # 移动到文件末尾 file.seek(0, 2) # 移动到文件末尾
file_size = file.tell() # 获取文件大小(字节) file_size = file.tell() # 获取文件大小(字节)