diff --git a/pywxdump/db/dbMicro.py b/pywxdump/db/dbMicro.py index 0f14448..7343f06 100644 --- a/pywxdump/db/dbMicro.py +++ b/pywxdump/db/dbMicro.py @@ -213,7 +213,8 @@ class MicroHandler(DatabaseBase): users[UserName] = { "wxid": UserName, "nickname": NickName, "remark": Remark, "account": Alias, "describe": describe, "headImgUrl": bigHeadImgUrl if bigHeadImgUrl else "", - "ExtraBuf": ExtraBuf, "LabelIDList": tuple(LabelIDList)} + "ExtraBuf": ExtraBuf, "LabelIDList": tuple(LabelIDList), + "extra": self.get_room_list(roomwxids=[UserName]).get(UserName, None)} return users @db_error @@ -241,7 +242,7 @@ class MicroHandler(DatabaseBase): sql = sql.replace(";", f"AND A.ChatRoomName LIKE '%{word}%' ;") if roomwxids: - sql = sql.replace(";", f"AND A.UserName IN ('" + "','".join(roomwxids) + "') ;") + sql = sql.replace(";", f"AND A.ChatRoomName IN ('" + "','".join(roomwxids) + "') ;") result = self.execute(sql) if not result: diff --git a/pywxdump/db/dbOpenIMContact.py b/pywxdump/db/dbOpenIMContact.py index 9be5627..c2a5e97 100644 --- a/pywxdump/db/dbOpenIMContact.py +++ b/pywxdump/db/dbOpenIMContact.py @@ -56,7 +56,7 @@ class OpenIMContactHandler(DatabaseBase): users[UserName] = { "wxid": UserName, "nickname": NickName, "remark": Remark, "account": UserName, "describe": '', "headImgUrl": BigHeadImgUrl if BigHeadImgUrl else "", - "ExtraBuf": None, "LabelIDList": tuple()} + "ExtraBuf": None, "LabelIDList": tuple(), "extra": None} return users