klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: Image upload size limit exceeded

Browse Source
This commit is contained in:
fit2cloud-chenyw
2025-12-30 10:37:40 +08:00
committed by fit2cloud-chenyw
parent 6d015ad0ad
commit b0150f6be5
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/system/api/assistant.py
+9 -2
View File
@@ -125,8 +125,15 @@ async def ui(session: SessionDep, data: str = Form(), files: List[UploadFile] =
file_name, flag_name = SQLBotFileUtils.split_filename_and_flag(origin_file_name)
file.filename = file_name
if flag_name == 'logo' or flag_name == 'float_icon':
SQLBotFileUtils.check_file(file=file, file_types=[".jpg", ".jpeg", ".png", ".svg"],
limit_file_size=(10 * 1024 * 1024))
try:
SQLBotFileUtils.check_file(file=file, file_types=[".jpg", ".jpeg", ".png", ".svg"],
limit_file_size=(10 * 1024 * 1024))
except ValueError as e:
error_msg = str(e)
if '文件大小超过限制' in error_msg:
raise ValueError(f"文件大小超过限制（最大 10 M）")
else:
raise e
if config_obj.get(flag_name):
SQLBotFileUtils.delete_file(config_obj.get(flag_name))
file_id = await SQLBotFileUtils.upload(file)