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-05-29 16:23:28 +08:00
parent 6f258d22ef
commit 5eacc7a86a
1 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
admin/dashboard/blueprints/contacts.py
View File

@@ -193,6 +193,16 @@ def api_contacts_update():
return jsonify({"success": False, "message": f"更新通讯录失败: {str(e)}"}), 500 return jsonify({"success": False, "message": f"更新通讯录失败: {str(e)}"}), 500
def get_or_create_eventloop():
"""获取或创建事件循环"""
try:
loop = asyncio.get_event_loop()
except RuntimeError:
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
return loop
@contacts_bp.route('/api/send_message', methods=['POST']) @contacts_bp.route('/api/send_message', methods=['POST'])
@login_required @login_required
def api_send_message(): def api_send_message():
@@ -219,9 +229,12 @@ def api_send_message():
if not server or not server.client: if not server or not server.client:
return jsonify({'success': False, 'message': '机器人未初始化'}) return jsonify({'success': False, 'message': '机器人未初始化'})
# 获取事件循环
loop = get_or_create_eventloop()
# 根据消息类型调用不同的发送方法 # 根据消息类型调用不同的发送方法
if msg_type == 'text': if msg_type == 'text':
client_msg_id, create_time, new_msg_id = asyncio.run( client_msg_id, create_time, new_msg_id = loop.run_until_complete(
server.client.send_text_message(wxid, content) server.client.send_text_message(wxid, content)
) )
return jsonify({ return jsonify({
@@ -237,7 +250,7 @@ def api_send_message():
if 'file' not in request.files: if 'file' not in request.files:
return jsonify({'success': False, 'message': '未上传文件'}) return jsonify({'success': False, 'message': '未上传文件'})
file = request.files['file'] file = request.files['file']
client_msg_id, create_time, new_msg_id = asyncio.run( client_msg_id, create_time, new_msg_id = loop.run_until_complete(
server.client.send_image_message(wxid, file.read()) server.client.send_image_message(wxid, file.read())
) )
return jsonify({ return jsonify({
@@ -253,7 +266,7 @@ def api_send_message():
if 'file' not in request.files: if 'file' not in request.files:
return jsonify({'success': False, 'message': '未上传文件'}) return jsonify({'success': False, 'message': '未上传文件'})
file = request.files['file'] file = request.files['file']
client_msg_id, create_time, new_msg_id = asyncio.run( client_msg_id, create_time, new_msg_id = loop.run_until_complete(
server.client.send_voice_message(wxid, file.read()) server.client.send_voice_message(wxid, file.read())
) )
return jsonify({ return jsonify({
@@ -269,7 +282,7 @@ def api_send_message():
if 'file' not in request.files: if 'file' not in request.files:
return jsonify({'success': False, 'message': '未上传文件'}) return jsonify({'success': False, 'message': '未上传文件'})
file = request.files['file'] file = request.files['file']
client_msg_id, new_msg_id = asyncio.run( client_msg_id, new_msg_id = loop.run_until_complete(
server.client.send_video_message(wxid, file.read()) server.client.send_video_message(wxid, file.read())
) )
return jsonify({ return jsonify({
@@ -284,7 +297,7 @@ def api_send_message():
url = content.get('url') url = content.get('url')
title = content.get('title', '') title = content.get('title', '')
description = content.get('description', '') description = content.get('description', '')
client_msg_id, create_time, new_msg_id = asyncio.run( client_msg_id, create_time, new_msg_id = loop.run_until_complete(
server.client.send_link_message(wxid, url, title, description) server.client.send_link_message(wxid, url, title, description)
) )
return jsonify({ return jsonify({