import threading from app.DataBase import msg_db, micro_msg_db, misc_db from app.person import Contact, Me, ContactDefault from app.DataBase.hard_link import parseBytesExtra, parseChatRoomData lock = threading.Lock() def singleton(cls): _instance = {} def inner(): if cls not in _instance: _instance[cls] = cls() return _instance[cls] return inner @singleton class PackageMsg: def __init__(self): self.ChatRoomMap = {} def get_package_message_all(self): """ 获取完整的聊天记录 """ updated_messages = [] # 用于存储修改后的消息列表 messages = msg_db.get_messages_all() for row in messages: row_list = list(row) # 删除不使用的几个字段 del row_list[13] del row_list[12] del row_list[11] del row_list[10] del row_list[9] strtalker = row[11] info = micro_msg_db.get_contact_by_username(strtalker) if info is not None: row_list.append(info[3]) row_list.append(info[4]) else: row_list.append("") row_list.append("") # 判断是否是群聊 if strtalker.__contains__("@chatroom"): # 自己发送 if row[4] == 1: row_list.append("我") else: # 存在BytesExtra为空的情况，此时消息类型应该为提示性消息。跳过不处理 if row[10] is None: continue # 解析BytesExtra msgbytes = parseBytesExtra(row[10]) wxid = "" for tmp in msgbytes[3]: if tmp[1] != 1: continue wxid = tmp[2] sender = "" # 获取群聊成员列表 membersMap = self.get_chatroom_member_list(strtalker) if membersMap is not None: if wxid in membersMap: sender = membersMap.get(wxid) else: senderinfo = micro_msg_db.get_contact_by_username(wxid) if senderinfo is not None: sender = senderinfo[4] membersMap[wxid] = senderinfo[4] if len(senderinfo[3]) > 0: sender = senderinfo[3] membersMap[wxid] = senderinfo[3] row_list.append(sender) else: if row[4] == 1: row_list.append("我") else: if info is not None: row_list.append(info[4]) else: row_list.append("") updated_messages.append(tuple(row_list)) return updated_messages def get_package_message_by_wxid(self, chatroom_wxid): """ 获取一个群聊的聊天记录 return list a[0]: localId, a[1]: talkerId, （和strtalker对应的，不是群聊信息发送人） a[2]: type, a[3]: subType, a[4]: is_sender, a[5]: timestamp, a[6]: status, （没啥用） a[7]: str_content, a[8]: str_time, （格式化的时间） a[9]: msgSvrId, a[10]: BytesExtra, a[11]: CompressContent, a[12]: msg_sender, （ContactPC 或 ContactDefault 类型，这个才是群聊里的信息发送人，不是群聊或者自己是发送者没有这个字段） """ updated_messages = [] # 用于存储修改后的消息列表 chatroom_members = self.get_chatroom_member_list(chatroom_wxid) messages = msg_db.get_messages(chatroom_wxid) for row in messages: message = list(row) if message[4] == 1: # 自己发送的就没必要解析了 message.append(Me()) updated_messages.append(message) continue if message[10] is None: # BytesExtra是空的跳过 message.append(ContactDefault(wxid)) updated_messages.append(message) continue msgbytes = parseBytesExtra(row[10]) wxid = "" for tmp in msgbytes[3]: if tmp[1] != 1: continue wxid = tmp[2] if wxid == "": # 系统消息里面 wxid 不存在 message.append(ContactDefault(wxid)) updated_messages.append(message) continue contact_info_list = micro_msg_db.get_contact_by_username(wxid) if contact_info_list is None: # 群聊中已退群的联系人不会保存在数据库里 message.append(ContactDefault(wxid)) updated_messages.append(message) continue contact_info = { "UserName": contact_info_list[0], "Alias": contact_info_list[1], "Type": contact_info_list[2], "Remark": contact_info_list[3], "NickName": contact_info_list[4], "smallHeadImgUrl": contact_info_list[7], } contact = Contact(contact_info) contact.smallHeadImgBLOG = misc_db.get_avatar_buffer(contact.wxid) contact.set_avatar(contact.smallHeadImgBLOG) message.append(contact) updated_messages.append(message) return updated_messages def get_chatroom_member_list(self, strtalker): membermap = {} """ 获取群聊成员 """ try: lock.acquire(True) if strtalker in self.ChatRoomMap: membermap = self.ChatRoomMap.get(strtalker) else: chatroom = micro_msg_db.get_chatroom_info(strtalker) if chatroom is None: return None # 解析RoomData数据 parsechatroom = parseChatRoomData(chatroom[1]) # 群成员数据放入字典存储 print(parsechatroom[7]) for mem in parsechatroom[1]: if mem[2] is not None and len(mem[2]) > 0: membermap[mem[1]] = mem[2] # wxid -> displayname self.ChatRoomMap[strtalker] = membermap finally: lock.release() return membermap if __name__ == "__main__": p = PackageMsg() print(p.get_package_message_by_wxid("18508451193@chatroom"))