fix

xaoyaoo · Aug 7, 2024 · 7f97882 · 7f97882
1 parent 5630ca4
commit 7f97882
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pywxdump/api/local_server.py b/pywxdump/api/local_server.py
@@ -176,7 +176,7 @@ def init_nokey():
     set_conf(g.caf, my_wxid, "wx_path", wx_path)
     set_conf(g.caf, my_wxid, "key", key)
     set_conf(g.caf, my_wxid, "my_wxid", my_wxid)
-    set_conf(g.caf, "test", "last", my_wxid)
+    set_conf(g.caf, g.at, "last", my_wxid)
     rdata = {
         "merge_path": merge_path,
         "wx_path": wx_path,

diff --git a/pywxdump/wx_core/wx_info.py b/pywxdump/wx_core/wx_info.py
@@ -389,7 +389,7 @@ def get_wx_db(msg_dir: str = None,
         db_types = None
 
     wxid_dirs = {}  # wx用户目录
-    if "All Users" in os.listdir(msg_dir) or "Applet" in os.listdir(msg_dir) or "WMPF" in os.listdir(msg_dir):
+    if wxids or "All Users" in os.listdir(msg_dir) or "Applet" in os.listdir(msg_dir) or "WMPF" in os.listdir(msg_dir):
         for sub_dir in os.listdir(msg_dir):
             if os.path.isdir(os.path.join(msg_dir, sub_dir)) and sub_dir not in ["All Users", "Applet", "WMPF"]:
                 wxid_dirs[os.path.basename(sub_dir)] = os.path.join(msg_dir, sub_dir)