liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

加入类型转换逻辑，防止解码失败

Browse Source
This commit is contained in:
liuwei
2025-06-12 11:51:39 +08:00
parent a10add9430
commit 7da746b0e0
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
admin/dashboard/blueprints/message_push.py
View File

@@ -407,7 +407,10 @@ def api_preview_task(task_id):
# 发送语音消息 # 发送语音消息
if task.get('content_voice'): if task.get('content_voice'):
send_message_in_thread(server.client.send_voice_message, recipient, Path(task['content_voice'])) voice_path = Path(task['content_voice'])
# 根据文件扩展名确定类型
voice_type = 'wav' if voice_path.suffix.lower() == '.wav' else 'mp3'
send_message_in_thread(server.client.send_voice_message, recipient, voice_path, voice_type)
# 发送视频消息 # 发送视频消息
if task.get('content_video'): if task.get('content_video'):
@@ -538,10 +541,12 @@ def upload_file():
file_type = 'video' file_type = 'video'
if file and allowed_file(file.filename, file_type): if file and allowed_file(file.filename, file_type):
# 生成安全的文件名 # 获取原始文件名和扩展名
filename = secure_filename(file.filename) original_filename = secure_filename(file.filename)
# 生成唯一文件名 file_ext = original_filename.rsplit('.', 1)[1].lower()
unique_filename = f"{uuid.uuid4().hex}_{filename}"
# 生成唯一文件名，保持原始扩展名
unique_filename = f"{uuid.uuid4().hex}.{file_ext}"
# 确保上传目录存在 # 确保上传目录存在
upload_folder = os.path.join(current_app.root_path, 'static', 'uploads') upload_folder = os.path.join(current_app.root_path, 'static', 'uploads')