微信私聊文件消息处理

app/endpoints/pipeline_endpoint.py

@@ -186,6 +186,7 @@ async def handle_add_messages_async(request: Request,token_id,msg,wxid):
         42: handle_name_card_async,
         43: handle_video_async,
         49: handle_xml_async, 
+        51: handle_file_message_async,
         37: handle_add_friend_notice_async,
         10002: handle_10002_msg_async,
         10000: handle_10000_msg_async
@@ -1098,7 +1099,7 @@ async def handle_name_card_async(request: Request,token_id,app_id, wxid,msg_data
         logger.error(f"未知错误: {e}")
 
 async def handle_video_async(request: Request,token_id,app_id, wxid,msg_data,from_wxid, to_wxid):
-    logger.info('视频消息')
+    logger.info(f'{wxid} 视频消息')
     try:
         msg_content_xml=msg_data["Content"]["string"]
         wx_video_url=await request.app.state.gewe_service.download_video_msg_async(token_id,app_id,msg_content_xml)
@@ -1243,6 +1244,23 @@ async def handle_xml_invite_group_async (request: Request,token_id,app_id, wxid,
         else:
             logger.warning(f'群聊邀请,同意加入群聊失败 {msg} {data}')   
 
+async def handle_file_message_async(request: Request,token_id,app_id, wxid,msg_data,from_wxid, to_wxid):
+    logger.info(f'{wxid} 微信文件消息处理')
+    try:
+        msg_content_xml=msg_data["Content"]["string"]
+        callback_to_user=from_wxid
+        wx_file_url=await request.app.state.gewe_service.download_file_msg_async(token_id,app_id,msg_content_xml)
+        if not wx_file_url:
+            logger.warning(f'处理微信文件消息异常')
+            return 
+        file_url=url_file_to_oss(wx_file_url)
+        wx_content_dialogue_message = [{"type": "file", "file_url": {"url":file_url}}]
+        k_message = dialogue_message(callback_to_user,wxid, wx_content_dialogue_message)
+        await request.app.state.kafka_service.send_message_async(k_message)
+        logger.info("发送对话 %s",k_message)        
+    except Exception as e:
+        logger.error(f"出现错误: {e}")
+
 async def handle_add_friend_notice_async(request: Request,token_id,app_id, wxid,msg_data,from_wxid, to_wxid):
     '''
     好友添加请求通知

services/gewe_service.py

@@ -708,6 +708,30 @@ class GeWeService:
                     logger.warning(f'下载视频失败 {response.status} {response.reason}')
                     return False        
 
+    async def download_file_msg_async(self, token_id: str, app_id: str,  xml: str):
+        data = {
+            "appId": app_id,
+            "xml": xml
+        }
+        headers = {
+            'X-GEWE-TOKEN': token_id,
+            'Content-Type': 'application/json'
+        }
+        url = f'{self.base_url}/v2/api/message/downloadFile'
+        async with aiohttp.ClientSession() as session:
+            async with session.post(url, json=data, headers=headers) as response:
+                if response.ok:
+                    data = await response.json()
+                    if data['ret'] == 200:
+                        return data['data']['fileUrl']
+                    else:
+                        logger.warning(f'下载视频失败 {data.get("ret", None),data.get("msg", None), data.get("data", None)}')
+                        return False
+                else:
+                    logger.warning(f'下载视频失败 {response.status} {response.reason}')
+                    return False        
+    
+
     ###############################  群模块   ###############################
     async def get_chatroom_info_async(self, token_id, app_id, chatroom_id):    
         '''