injectlib/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2025-11-25 11:29:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: split WeCom callback handlers

Browse Source
This commit is contained in:
Alfonsxh
2025-10-28 18:30:55 +08:00
committed by Junyan Qin
parent 2dd2abedde
commit 4a02c531b2
1 changed files with 53 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
libs/wecom_ai_bot_api/api.py
View File

@@ -295,7 +295,7 @@ class WecomBotClient:
except Exception: except Exception:
await self.logger.error(traceback.format_exc()) await self.logger.error(traceback.format_exc())
async def _handle_initial_message(self, msg_json: dict[str, Any], nonce: str) -> tuple[Response, int]: async def _handle_post_initial_response(self, msg_json: dict[str, Any], nonce: str) -> tuple[Response, int]:
"""处理企业微信首次推送的消息，返回 stream_id 并开启流水线。 """处理企业微信首次推送的消息，返回 stream_id 并开启流水线。
Args: Args:
@@ -324,7 +324,7 @@ class WecomBotClient:
payload = self._build_stream_payload(session.stream_id, '', False) payload = self._build_stream_payload(session.stream_id, '', False)
return await self._encrypt_and_reply(payload, nonce) return await self._encrypt_and_reply(payload, nonce)
async def _handle_stream_refresh(self, msg_json: dict[str, Any], nonce: str) -> tuple[Response, int]: async def _handle_post_followup_response(self, msg_json: dict[str, Any], nonce: str) -> tuple[Response, int]:
"""处理企业微信的流式刷新请求，按需返回增量片段。 """处理企业微信的流式刷新请求，按需返回增量片段。
Args: Args:
@@ -375,7 +375,20 @@ class WecomBotClient:
await self.logger.info(f'{request.method} {request.url} {str(request.args)}') await self.logger.info(f'{request.method} {request.url} {str(request.args)}')
if request.method == 'GET': if request.method == 'GET':
# GET 用于验证回调 URL，有效期内直接返回微信给的 echostr return await self._handle_get_callback()
if request.method == 'POST':
return await self._handle_post_callback()
return Response('', status=405)
except Exception:
await self.logger.error(traceback.format_exc())
return Response('Internal Server Error', status=500)
async def _handle_get_callback(self) -> tuple[Response, int] | Response:
"""处理企业微信的 GET 验证请求。"""
msg_signature = unquote(request.args.get('msg_signature', '')) msg_signature = unquote(request.args.get('msg_signature', ''))
timestamp = unquote(request.args.get('timestamp', '')) timestamp = unquote(request.args.get('timestamp', ''))
nonce = unquote(request.args.get('nonce', '')) nonce = unquote(request.args.get('nonce', ''))
@@ -392,8 +405,8 @@ class WecomBotClient:
return Response(decrypted_str, mimetype='text/plain') return Response(decrypted_str, mimetype='text/plain')
if request.method != 'POST': async def _handle_post_callback(self) -> tuple[Response, int] | Response:
return Response('', status=405) """处理企业微信的 POST 回调请求。"""
self.stream_sessions.cleanup() self.stream_sessions.cleanup()
@@ -416,15 +429,9 @@ class WecomBotClient:
msg_json = json.loads(decrypted_xml) msg_json = json.loads(decrypted_xml)
if msg_json.get('msgtype') == 'stream': if msg_json.get('msgtype') == 'stream':
# 企业微信刷新请求：尝试从队列中取出增量回复 return await self._handle_post_followup_response(msg_json, nonce)
return await self._handle_stream_refresh(msg_json, nonce)
# 首次请求：快速返回 stream_id 并异步处理流水线 return await self._handle_post_initial_response(msg_json, nonce)
return await self._handle_initial_message(msg_json, nonce)
except Exception:
await self.logger.error(traceback.format_exc())
return Response('Internal Server Error', status=500)
async def get_message(self, msg_json): async def get_message(self, msg_json):
message_data = {} message_data = {}