Skip to content

Commit 0ea0e75

Browse files
fix: Image upload size limit exceeded
1 parent ad0b747 commit 0ea0e75

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

backend/apps/system/api/assistant.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,15 @@ async def ui(session: SessionDep, data: str = Form(), files: List[UploadFile] =
125125
file_name, flag_name = SQLBotFileUtils.split_filename_and_flag(origin_file_name)
126126
file.filename = file_name
127127
if flag_name == 'logo' or flag_name == 'float_icon':
128-
SQLBotFileUtils.check_file(file=file, file_types=[".jpg", ".jpeg", ".png", ".svg"],
129-
limit_file_size=(10 * 1024 * 1024))
128+
try:
129+
SQLBotFileUtils.check_file(file=file, file_types=[".jpg", ".jpeg", ".png", ".svg"],
130+
limit_file_size=(10 * 1024 * 1024))
131+
except ValueError as e:
132+
error_msg = str(e)
133+
if '文件大小超过限制' in error_msg:
134+
raise ValueError(f"文件大小超过限制(最大 10 M)")
135+
else:
136+
raise e
130137
if config_obj.get(flag_name):
131138
SQLBotFileUtils.delete_file(config_obj.get(flag_name))
132139
file_id = await SQLBotFileUtils.upload(file)

0 commit comments

Comments
 (0)