diff --git a/pywxdump/api/local_server.py b/pywxdump/api/local_server.py index c152fbb9..d14a368b 100644 --- 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 index e383639d..24c40718 100644 --- 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)