import datetime import json import os import re import time import pilk from bridge.context import ContextType from channel.chat_message import ChatMessage from common.log import logger from ntwork.const import send_type def get_with_retry(get_func, max_retries=5, delay=5): retries = 0 result = None while retries < max_retries: result = get_func() if result: break logger.warning(f"获取数据失败，重试第{retries + 1}次······") retries += 1 time.sleep(delay) # 等待一段时间后重试 return result def get_room_info(wework, conversation_id): logger.debug(f"传入的 conversation_id: {conversation_id}") rooms = wework.get_rooms() if not rooms or 'room_list' not in rooms: logger.error(f"获取群聊信息失败: {rooms}") return None time.sleep(1) logger.debug(f"获取到的群聊信息: {rooms}") for room in rooms['room_list']: if room['conversation_id'] == conversation_id: return room return None def cdn_download(wework, message, file_name): data = message["data"] aes_key = data["cdn"]["aes_key"] file_size = data["cdn"]["size"] # 获取当前工作目录，然后与文件名拼接得到保存路径 current_dir = os.getcwd() save_path = os.path.join(current_dir, "tmp", file_name) # 下载保存图片到本地 if "url" in data["cdn"].keys() and "auth_key" in data["cdn"].keys(): url = data["cdn"]["url"] auth_key = data["cdn"]["auth_key"] # result = wework.wx_cdn_download(url, auth_key, aes_key, file_size, save_path) # ntwork库本身接口有问题，缺失了aes_key这个参数 """ 下载wx类型的cdn文件，以https开头 """ data = { 'url': url, 'auth_key': auth_key, 'aes_key': aes_key, 'size': file_size, 'save_path': save_path } result = wework._WeWork__send_sync(send_type.MT_WXCDN_DOWNLOAD_MSG, data) # 直接用wx_cdn_download的接口内部实现来调用 elif "file_id" in data["cdn"].keys(): if message["type"] == 11042: file_type = 2 elif message["type"] == 11045: file_type = 5 file_id = data["cdn"]["file_id"] result = wework.c2c_cdn_download(file_id, aes_key, file_size, file_type, save_path) else: logger.error(f"something is wrong, data: {data}") return # 输出下载结果 logger.debug(f"result: {result}") def c2c_download_and_convert(wework, message, file_name): data = message["data"] aes_key = data["cdn"]["aes_key"] file_size = data["cdn"]["size"] file_type = 5 file_id = data["cdn"]["file_id"] current_dir = os.getcwd() save_path = os.path.join(current_dir, "tmp", file_name) result = wework.c2c_cdn_download(file_id, aes_key, file_size, file_type, save_path) logger.debug(result) # 在下载完SILK文件之后，立即将其转换为WAV文件 base_name, _ = os.path.splitext(save_path) wav_file = base_name + ".wav" pilk.silk_to_wav(save_path, wav_file, rate=24000) # 删除SILK文件 try: os.remove(save_path) except Exception as e: pass class WeworkMessage(ChatMessage): def __init__(self, wework_msg, wework, is_group=False): try: super().__init__(wework_msg) self.msg_id = wework_msg['data'].get('conversation_id', wework_msg['data'].get('room_conversation_id')) # 使用.get()防止 'send_time' 键不存在时抛出错误 self.create_time = wework_msg['data'].get("send_time") self.is_group = is_group self.wework = wework if wework_msg["type"] == 11041: # 文本消息类型 if any(substring in wework_msg['data']['content'] for substring in ("该消息类型暂不能展示", "不支持的消息类型")): return self.ctype = ContextType.TEXT self.content = wework_msg['data']['content'] elif wework_msg["type"] == 11044: # 语音消息类型，需要缓存文件 file_name = datetime.datetime.now().strftime('%Y%m%d%H%M%S') + ".silk" base_name, _ = os.path.splitext(file_name) file_name_2 = base_name + ".wav" current_dir = os.getcwd() self.ctype = ContextType.VOICE self.content = os.path.join(current_dir, "tmp", file_name_2) self._prepare_fn = lambda: c2c_download_and_convert(wework, wework_msg, file_name) elif wework_msg["type"] == 11042: # 图片消息类型，需要下载文件 file_name = datetime.datetime.now().strftime('%Y%m%d%H%M%S') + ".jpg" current_dir = os.getcwd() self.ctype = ContextType.IMAGE self.content = os.path.join(current_dir, "tmp", file_name) self._prepare_fn = lambda: cdn_download(wework, wework_msg, file_name) elif wework_msg["type"] == 11045: # 文件消息 print("文件消息") print(wework_msg) file_name = datetime.datetime.now().strftime('%Y%m%d%H%M%S') file_name = file_name + wework_msg['data']['cdn']['file_name'] current_dir = os.getcwd() self.ctype = ContextType.FILE self.content = os.path.join(current_dir, "tmp", file_name) self._prepare_fn = lambda: cdn_download(wework, wework_msg, file_name) elif wework_msg["type"] == 11047: # 链接消息 self.ctype = ContextType.SHARING self.content = wework_msg['data']['url'] elif wework_msg["type"] == 11072: # 新成员入群通知 self.ctype = ContextType.JOIN_GROUP member_list = wework_msg['data']['member_list'] self.actual_user_nickname = member_list[0]['name'] self.actual_user_id = member_list[0]['user_id'] self.content = f"{self.actual_user_nickname}加入了群聊！" directory = os.path.join(os.getcwd(), "tmp") rooms = get_with_retry(wework.get_rooms) if not rooms: logger.error("更新群信息失败···") else: result = {} for room in rooms['room_list']: # 获取聊天室ID room_wxid = room['conversation_id'] # 获取聊天室成员 room_members = wework.get_room_members(room_wxid) # 将聊天室成员保存到结果字典中 result[room_wxid] = room_members with open(os.path.join(directory, 'wework_room_members.json'), 'w', encoding='utf-8') as f: json.dump(result, f, ensure_ascii=False, indent=4) logger.info("有新成员加入，已自动更新群成员列表缓存！") else: raise NotImplementedError( "Unsupported message type: Type:{} MsgType:{}".format(wework_msg["type"], wework_msg["MsgType"])) data = wework_msg['data'] login_info = self.wework.get_login_info() logger.debug(f"login_info: {login_info}") nickname = f"{login_info['username']}({login_info['nickname']})" if login_info['nickname'] else login_info['username'] user_id = login_info['user_id'] sender_id = data.get('sender') conversation_id = data.get('conversation_id') sender_name = data.get("sender_name") self.from_user_id = user_id if sender_id == user_id else conversation_id self.from_user_nickname = nickname if sender_id == user_id else sender_name self.to_user_id = user_id self.to_user_nickname = nickname self.other_user_nickname = sender_name self.other_user_id = conversation_id if self.is_group: conversation_id = data.get('conversation_id') or data.get('room_conversation_id') self.other_user_id = conversation_id if conversation_id: room_info = get_room_info(wework=wework, conversation_id=conversation_id) self.other_user_nickname = room_info.get('nickname', None) if room_info else None self.from_user_nickname = room_info.get('nickname', None) if room_info else None at_list = data.get('at_list', []) tmp_list = [] for at in at_list: tmp_list.append(at['nickname']) at_list = tmp_list logger.debug(f"at_list: {at_list}") logger.debug(f"nickname: {nickname}") self.is_at = False if nickname in at_list or login_info['nickname'] in at_list or login_info['username'] in at_list: self.is_at = True self.at_list = at_list # 检查消息内容是否包含@用户名。处理复制粘贴的消息，这类消息可能不会触发@通知，但内容中可能包含 "@用户名"。 content = data.get('content', '') name = nickname pattern = f"@{re.escape(name)}(\u2005|\u0020)" if re.search(pattern, content): logger.debug(f"Wechaty message {self.msg_id} includes at") self.is_at = True if not self.actual_user_id: self.actual_user_id = data.get("sender") self.actual_user_nickname = sender_name if self.ctype != ContextType.JOIN_GROUP else self.actual_user_nickname else: logger.error("群聊消息中没有找到 conversation_id 或 room_conversation_id") logger.debug(f"WeworkMessage has been successfully instantiated with message id: {self.msg_id}") except Exception as e: logger.error(f"在 WeworkMessage 的初始化过程中出现错误：{e}") raise e