From 7a74fee463d6057bb8116b6ec128a3ac44a3e20a Mon Sep 17 00:00:00 2001
From: xaoyaoo <xaoyaoo@qq.com>
Date: Thu, 11 Jan 2024 11:11:31 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=B8=93=E4=B8=9A=E5=B7=A5?=
 =?UTF-8?q?=E5=85=B7=E4=B8=AD=E8=8E=B7=E5=8F=96=E6=AD=A3=E5=9C=A8=E8=BF=90?=
 =?UTF-8?q?=E8=A1=8C=E5=BE=AE=E4=BF=A1=E4=BF=A1=E6=81=AF=E5=86=85=E5=AE=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pywxdump/api/api.py             | 14 ++++++++++++++
 pywxdump/wx_info/get_wx_info.py |  5 ++++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/pywxdump/api/api.py b/pywxdump/api/api.py
index e11bab9..8333d6a 100644
--- a/pywxdump/api/api.py
+++ b/pywxdump/api/api.py
@@ -11,6 +11,7 @@ import os
 from flask import Flask, request, render_template, g, Blueprint, send_file
 from pywxdump import analyzer, read_img_dat, read_audio
 from pywxdump.api.rjson import ReJson, RqJson
+from pywxdump import read_info,VERSION_LIST
 
 # app = Flask(__name__, static_folder='../ui/web/dist', static_url_path='/')
 
@@ -186,6 +187,19 @@ def get_audio():
     return ReJson(0, video_data)
 
 
+# 这部分为专业工具的api
+@api.route('/api/wxinfo', methods=["GET", 'POST'])
+def get_wxinfo():
+    """
+    获取微信信息
+    :return:
+    """
+    import pythoncom
+    pythoncom.CoInitialize()
+    wxinfos = read_info(VERSION_LIST)
+    pythoncom.CoUninitialize()
+    return ReJson(0, wxinfos)
+
 @api.route('/')
 def index():
     return render_template('index.html')
diff --git a/pywxdump/wx_info/get_wx_info.py b/pywxdump/wx_info/get_wx_info.py
index 892431e..69c5e29 100644
--- a/pywxdump/wx_info/get_wx_info.py
+++ b/pywxdump/wx_info/get_wx_info.py
@@ -140,7 +140,10 @@ def get_key(pid, db_path, addr_len):
 
 
 # 读取微信信息(account,mobile,name,mail,wxid,key)
-def read_info(version_list: dict, is_logging: bool = False, save_path: str = None):
+def read_info(version_list: dict = None, is_logging: bool = False, save_path: str = None):
+    if version_list is None:
+        version_list = {}
+
     wechat_process = []
     result = []
     error = ""