增加了单api的运行和返回聊天中所有用户接口 (#73)

add msg only api、add 单api运行方式、增加返回会话所有用户信息接口

pywxdump/analyzer/export_chat.py

@@ -24,6 +24,30 @@ from .utils import get_md5, attach_databases, execute_sql, get_type_name, match_
 from .db_parsing import parse_xml_string, decompress_CompressContent, read_BytesExtra
 
 
+def get_contact(MicroMsg_db_path,wx_id):
+    """
+    获取联系人信息
+    :param MicroMsg_db_path: MicroMsg.db 文件路径
+    :param wx_id: 微信id
+    :return: 联系人信息
+    """
+    db = sqlite3.connect(MicroMsg_db_path)
+    cursor = db.cursor()
+    # 获取username是wx_id的用户
+    sql = ("SELECT A.UserName, A.NickName, A.Remark,A.Alias,A.Reserved6,B.bigHeadImgUrl "
+           "FROM Contact A,ContactHeadImgUrl B "
+           f"WHERE A.UserName = '{wx_id}' AND A.UserName = B.usrName "
+           "ORDER BY NickName ASC;")
+    cursor.execute(sql)
+    result = cursor.fetchone()
+    print('联系人信息：', result)
+    if not result:
+        print('居然没找到！')
+        print(wx_id)
+        return None
+    return {"username": result[0], "nickname": result[1], "remark": result[2], "account": result[3], "describe": result[4], "headImgUrl": result[5]}
+    
+
 def get_contact_list(MicroMsg_db_path):
     """
     获取联系人列表
@@ -79,6 +103,47 @@ def get_chatroom_list(MicroMsg_db_path):
              "Announcement": Announcement, "AnnouncementEditor": AnnouncementEditor})
     return rooms
 
+def get_room_user_list(MSG_db_path, selected_talker):
+    """
+    获取群聊中包含的所有用户列表
+    :param MSG_db_path: MSG.db 文件路径
+    :param selected_talker: 选中的聊天对象 wxid
+    :return: 聊天用户列表
+    """
+    
+    # 连接 MSG_ALL.db 数据库，并执行查询
+    db1 = sqlite3.connect(MSG_db_path)
+    cursor1 = db1.cursor()
+
+    sql = (
+        "SELECT localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType,CreateTime,MsgSvrID,DisplayContent,CompressContent,BytesExtra,ROW_NUMBER() OVER (ORDER BY CreateTime ASC) AS id "
+        "FROM MSG WHERE StrTalker=? "
+        "ORDER BY CreateTime ASC")
+
+    cursor1.execute(sql, (selected_talker,))
+    result1 = cursor1.fetchall()
+    cursor1.close()
+    db1.close()
+    user_list = []
+    read_user_wx_id = []
+    for row  in result1:
+        localId, IsSender, StrContent, StrTalker, Sequence, Type, SubType, CreateTime, MsgSvrID, DisplayContent, CompressContent, BytesExtra, id = row
+        bytes_extra = read_BytesExtra(BytesExtra)
+        if bytes_extra:
+            try:
+                talker = bytes_extra['3'][0]['2'].decode('utf-8', errors='ignore')
+            except:
+                continue
+        if talker in read_user_wx_id:
+            continue
+        user = get_contact(MSG_db_path, talker)
+        if not user:
+            continue
+        user_list.append(user)
+        read_user_wx_id.append(talker)
+    return user_list
+            
+
 
 def get_msg_list(MSG_db_path, selected_talker="", start_index=0, page_size=500):
     """

pywxdump/api/api.py

@@ -13,6 +13,7 @@ import shutil
 
 from flask import Flask, request, render_template, g, Blueprint, send_file, make_response, session
 from pywxdump import analyzer, read_img_dat, read_audio, get_wechat_db, get_core_db
+from pywxdump.analyzer.export_chat import get_contact, get_room_user_list
 from pywxdump.api.rjson import ReJson, RqJson
 from pywxdump.api.utils import read_session, save_session, error9999
 from pywxdump import read_info, VERSION_LIST, batch_decrypt, BiasAddr, merge_db, decrypt_merge
@@ -190,6 +191,45 @@ def contact_count_list():
     except Exception as e:
         return ReJson(9999, msg=str(e))
 
+@api.route('/api/msgs_user_list',methods=['GET','POST'])
+@error9999
+def get_msg_user_list():
+    msg_path = request.headers.get("msg_path")
+    micro_path = request.headers.get("micro_path")
+    if not msg_path:
+        msg_path = read_session(g.sf, "msg_path")
+    if not micro_path:
+        micro_path = read_session(g.sf, "micro_path")
+    wxid = request.json.get("wxid")
+    # msg_list = analyzer.get_msg_list(msg_path, wxid, start_index=start, page_size=limit)
+    my_wxid = read_session(g.sf, "my_wxid")
+    userlist = []
+    if wxid.endswith("@chatroom"):
+        # 群聊
+        userlist = get_room_user_list(msg_path, wxid)
+    else:
+        # 单聊
+        user = get_contact(micro_path, wxid)
+        my_user = get_contact(micro_path,my_wxid)
+        userlist.append(user)
+        userlist.append(my_user)
+    return ReJson(0, {"user_list":userlist})
+
+@api.route('/api/msgs_list',methods=['GET','POST'])
+@error9999
+def get_msg_list():
+    msg_path = request.headers.get("msg_path")
+    micro_path = request.headers.get("micro_path")
+    if not msg_path:
+        msg_path = read_session(g.sf, "msg_path")
+    if not micro_path:
+        micro_path = read_session(g.sf, "micro_path")
+    start = request.json.get("start")
+    limit = request.json.get("limit")
+    wxid = request.json.get("wxid")
+    my_wxid = read_session(g.sf, "my_wxid")
+    msg_list = analyzer.get_msg_list(msg_path, wxid, start_index=start, page_size=limit)
+    return ReJson(0, {"msg_list":msg_list,'my_wxid':my_wxid})
 
 @api.route('/api/msgs', methods=["GET", 'POST'])
 @error9999

pywxdump/cli.py

@@ -412,6 +412,26 @@ class MainUi():
 
         start_falsk(port=port, online=online, debug=debug)
 
+class MainApi():
+    def init_parses(self, parser):
+        self.mode = "api"
+        # 添加 'api' 子命令解析器
+        sb_api = parser.add_parser(self.mode, help="启动api")
+        sb_api.add_argument("-p", '--port', metavar="", type=int, help="(可选)端口号", default=5000)
+        sb_api.add_argument("--online", type=bool, help="(可选)是否在线查看(局域网查看)", required=False, default=False,
+                           metavar="")
+        sb_api.add_argument("--debug", type=bool, help="(可选)是否开启debug模式", default=False)
+        return sb_api
+
+    def run(self, args):
+        print(f"[*] PyWxDump v{pywxdump.__version__}")
+        # 从命令行参数获取值
+        online = args.online
+        port = args.port
+        debug = args.debug
+
+        start_falsk(port=port, online=online, debug=debug,isopenBrowser=False)
+
 
 class CustomArgumentParser(argparse.ArgumentParser):
     def format_help(self):
@@ -487,11 +507,16 @@ def console_run():
     sb_ui = main_ui.init_parses(subparsers)
     modes[main_ui.mode] = main_ui
 
+     # 添加 'api' 子命令解析器
+    main_api = MainApi()
+    sb_api = main_api.init_parses(subparsers)
+    modes[main_api.mode] = main_api
+
     # 检查是否需要显示帮助信息
     if len(sys.argv) == 1:
         sys.argv.append('ui')
     elif len(sys.argv) == 2 and sys.argv[1] in modes.keys() and sys.argv[1] not in [main_all.mode, main_wx_info.mode,
-                                                                                    main_wx_db_path.mode, main_ui.mode]:
+                                                                                    main_wx_db_path.mode, main_ui.mode ,main_api.mode]:
         sys.argv.append('-h')
 
     args = parser.parse_args()  # 解析命令行参数