mirror of https://github.com/jackwener/wx-cli.git
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
"""
|
|
配置加载器 - 从 config.json 读取路径配置
|
|
首次运行时自动生成 config.json 模板
|
|
"""
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
CONFIG_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "config.json")
|
|
|
|
_DEFAULT = {
|
|
"db_dir": r"D:\xwechat_files\your_wxid\db_storage",
|
|
"keys_file": "all_keys.json",
|
|
"decrypted_dir": "decrypted",
|
|
"wechat_process": "Weixin.exe",
|
|
}
|
|
|
|
|
|
def load_config():
|
|
if not os.path.exists(CONFIG_FILE):
|
|
with open(CONFIG_FILE, "w") as f:
|
|
json.dump(_DEFAULT, f, indent=4)
|
|
print(f"[!] 已生成配置文件: {CONFIG_FILE}")
|
|
print(" 请修改 config.json 中的路径后重新运行")
|
|
sys.exit(1)
|
|
|
|
with open(CONFIG_FILE) as f:
|
|
cfg = json.load(f)
|
|
|
|
# 将相对路径转为绝对路径
|
|
base = os.path.dirname(os.path.abspath(__file__))
|
|
for key in ("keys_file", "decrypted_dir"):
|
|
if key in cfg and not os.path.isabs(cfg[key]):
|
|
cfg[key] = os.path.join(base, cfg[key])
|
|
|
|
return cfg
|