feat: tqdm progress bar + setup.py wizard

export_all_chats.py

@@ -29,6 +29,13 @@ from contextlib import closing
 from datetime import datetime
 
 import mcp_server
+
+# 尝试导入 tqdm 作为进度条（可选）
+try:
+    from tqdm import tqdm as _tqdm
+except ImportError:
+    _tqdm = None
+
 from chat_export_helpers import _extract_content, _msg_type_str, _resolve_sender
 
 
@@ -215,7 +222,9 @@ def main():
 
     t0 = time.time()
     ok, skip, err, total = 0, 0, 0, 0
-    for i, username in enumerate(sessions, 1):
+
+    iterable = _tqdm(sessions, desc="导出进度") if _tqdm else sessions
+    for i, username in enumerate(iterable, 1):
         display = names.get(username, username)
         success, count, reason = export_one(
             username, output_dir, names, transcribe=args.with_transcriptions
@@ -223,21 +232,26 @@ def main():
         if success:
             ok += 1
             total += count
-            if i <= 10 or i % 100 == 0:
+            if not _tqdm:
-                elapsed = time.time() - t0
+                if i <= 10 or i % 100 == 0:
-                eta = (elapsed / i) * (len(sessions) - i) if i > 0 else 0
+                    elapsed = time.time() - t0
-                print(
+                    eta = (elapsed / i) * (len(sessions) - i) if i > 0 else 0
-                    f"[{i}/{len(sessions)}] {display} - {count} 条消息"
+                    print(
-                    + (f"  ETA {eta/60:.0f}分" if i > 1 else "")
+                        f"[{i}/{len(sessions)}] {display} - {count} 条消息"
-                )
+                        + (f"  ETA {eta/60:.0f}分" if i > 1 else "")
+                    )
         else:
             if "no tables" in str(reason) or "empty" in str(reason):
                 skip += 1
-                if i <= 10 or i % 50 == 0:
+                if not _tqdm:
-                    print(f"[{i}/{len(sessions)}] {display} - 跳过({reason})")
+                    if i <= 10 or i % 50 == 0:
+                        print(f"[{i}/{len(sessions)}] {display} - 跳过({reason})")
             else:
                 err += 1
-                print(f"[{i}/{len(sessions)}] {display} - 失败: {reason}")
+                if not _tqdm:
+                    print(f"[{i}/{len(sessions)}] {display} - 失败: {reason}")
+                elif _tqdm:
+                    _tqdm.write(f"失败: {display} - {reason}")
 
     elapsed = time.time() - t0
     print()

requirements.txt

@@ -1,3 +1,4 @@
 pycryptodome>=3.19,<4
 zstandard>=0.22,<1
 mcp>=1.0,<2
+# 可选：进度条 (pip install tqdm)

setup.py

@@ -0,0 +1,245 @@
+#!/usr/bin/env python3
+"""
+WeChat Decrypt — 交互式配置向导
+
+检测微信数据目录、选择转录 backend、生成 config.json。
+
+用法:
+    python3 setup.py           # 交互式配置
+    python3 setup.py --check   # 仅检查环境，不修改文件
+"""
+
+import argparse
+import glob
+import json
+import os
+import platform
+import shutil
+import subprocess
+import sys
+
+
+def detect_wechat_dir():
+    """自动检测微信数据目录"""
+    system = platform.system().lower()
+
+    if system == "darwin":
+        containers = (
+            os.path.expanduser(
+                "~/Library/Containers/com.tencent.xinWeChat/Data/Documents/xwechat_files"
+            )
+        )
+        if os.path.exists(containers):
+            dirs = [
+                os.path.join(containers, d, "db_storage")
+                for d in os.listdir(containers)
+                if os.path.isdir(os.path.join(containers, d))
+            ]
+            dirs = [d for d in dirs if os.path.exists(d)]
+            # 按修改时间排序，最近活跃的排最前
+            dirs.sort(key=lambda d: os.path.getmtime(d), reverse=True)
+            return dirs if dirs else None
+
+    elif system == "linux":
+        home = os.path.expanduser("~")
+        candidates = [
+            os.path.join(home, "Documents", "xwechat_files"),
+            os.path.join(home, ".xwechat", "files"),
+        ]
+        for c in candidates:
+            if os.path.exists(c):
+                dirs = [
+                    os.path.join(c, d, "db_storage")
+                    for d in os.listdir(c)
+                    if os.path.isdir(os.path.join(c, d))
+                ]
+                dirs = [d for d in dirs if os.path.exists(d)]
+                if dirs:
+                    return dirs
+
+    elif system == "windows":
+        localappdata = os.environ.get("LOCALAPPDATA", "")
+        candidates = [
+            os.path.join(localappdata, "xwechat_files"),
+            os.path.join(os.environ.get("USERPROFILE", ""), "Documents", "xwechat_files"),
+        ]
+        for c in candidates:
+            if os.path.exists(c):
+                dirs = [
+                    os.path.join(c, d, "db_storage")
+                    for d in os.listdir(c)
+                    if os.path.isdir(os.path.join(c, d))
+                ]
+                dirs = [d for d in dirs if os.path.exists(d)]
+                if dirs:
+                    return dirs
+
+    return None
+
+
+def detect_transcription_backends():
+    """检测可用的转录 backend"""
+    backends = {"local": True}  # local 总是可用（如果安装了依赖）
+
+    # whisper-cpp
+    if shutil.which("whisper-cpp") or shutil.which("whisper-cli"):
+        backends["whisper_cpp"] = True
+        model_dirs = [
+            os.path.expanduser("~/Library/Application Support/whisper-cpp"),
+            os.path.expanduser("~/whisper-models"),
+            "/opt/homebrew/share/whisper-cpp/models",
+        ]
+        for md in model_dirs:
+            if os.path.exists(md):
+                models = [f for f in os.listdir(md) if f.startswith("ggml-") and f.endswith(".bin")]
+                if models:
+                    backends["whisper_cpp_model"] = models[0]
+                    break
+    else:
+        backends["whisper_cpp"] = False
+
+    # openai
+    try:
+        import openai  # noqa
+        backends["openai"] = True
+    except ImportError:
+        backends["openai"] = False
+
+    return backends
+
+
+def check_environment():
+    """检查环境，返回状态信息"""
+    print("=== 环境检查 ===")
+    py_ver = sys.version.split()[0]
+    print(f"[Python] {py_ver}")
+
+    # venv
+    in_venv = sys.prefix != sys.base_prefix
+    print(f"[venv]  {'是' if in_venv else '否'}")
+    if not in_venv:
+        print("       建议使用: python3 -m venv .venv && source .venv/bin/activate")
+
+    # whisper-cpp
+    whisper_bin = shutil.which("whisper-cpp") or shutil.which("whisper-cli")
+    print(f"[whisper-cpp] {'✓ ' + whisper_bin if whisper_bin else '✗ 未安装 (brew install whisper-cpp)'}")
+
+    # config
+    if os.path.exists("config.json"):
+        with open("config.json") as f:
+            cfg = json.load(f)
+        db_dir = cfg.get("db_dir", "?")
+        backend = cfg.get("transcription_backend", "未设置")
+        print(f"[config.json] ✓ (db_dir = {db_dir}, backend = {backend})")
+    else:
+        print("[config.json] ✗ 未找到")
+
+    # 微信目录
+    dirs = detect_wechat_dir()
+    if dirs:
+        print(f"[微信目录] 找到 {len(dirs)} 个:")
+        for d in dirs:
+            age_days = (os.path.getmtime(__file__ if '__file__' in dir() else 0) - os.path.getmtime(d)) / 86400 if os.path.exists(d) else 0
+            print(f"            {d}")
+    else:
+        print("[微信目录] 未找到自动检测路径")
+
+    return dirs
+
+
+def interactive_setup():
+    """交互式配置向导"""
+    print("\n=== 微信解密工具 — 配置向导 ===\n")
+
+    # 加载或创建配置
+    config = {}
+    if os.path.exists("config.json"):
+        with open("config.json") as f:
+            config = json.load(f)
+        print(f"现有 config.json 已加载 ({len(config)} 个字段)")
+        print()
+
+    # 微信数据目录
+    detected = detect_wechat_dir()
+    if detected:
+        if len(detected) == 1:
+            chosen = detected[0]
+            print(f"[1/3] 微信数据目录: 自动检测到")
+            print(f"      {chosen}")
+        else:
+            print(f"[1/3] 检测到 {len(detected)} 个微信数据目录:")
+            for i, d in enumerate(detected, 1):
+                print(f"      [{i}] {d}")
+            try:
+                sel = int(input("\n请选择 (1-{}): ".format(len(detected))) or "1")
+                chosen = detected[sel - 1]
+            except (ValueError, IndexError):
+                chosen = detected[0]
+        config["db_dir"] = chosen
+    else:
+        print("[1/3] 微信数据目录: 未能自动检测")
+        default_path = os.path.expanduser("~/Documents/xwechat_files/your_wxid/db_storage")
+        chosen = input(f"      请手动输入路径 [{default_path}]: ") or default_path
+        config["db_dir"] = chosen
+
+    # 转录 backend
+    backends = detect_transcription_backends()
+    print(f"\n[2/3] 语音转录 backend:")
+    print(f"      [1] local — 本地 CPU 转录（默认，隐私最佳，速度较慢）")
+    status_w = "✓" if backends.get("whisper_cpp") else "✗ (brew install whisper-cpp)"
+    print(f"      [2] whisper_cpp — GPU 加速 ({status_w})")
+    status_o = "✓" if backends.get("openai") else "✗ (pip install openai)"
+    print(f"      [3] openai — API 转录 ({status_o})")
+
+    try:
+        sel = int(input("\n      请选择 (1-3) [1]: ") or "1")
+        if sel == 2:
+            config["transcription_backend"] = "whisper_cpp"
+            if "whisper_cpp_model" in backends:
+                config["whisper_cpp_model"] = backends["whisper_cpp_model"]
+        elif sel == 3:
+            config["transcription_backend"] = "openai"
+            key = input("      输入 OpenAI API Key: ").strip()
+            if key:
+                config["openai_api_key"] = key
+        else:
+            config["transcription_backend"] = "local"
+    except (ValueError, IndexError):
+        config["transcription_backend"] = "local"
+
+    # 确认
+    print(f"\n[3/3] 即将写入 config.json:")
+    print(json.dumps(config, indent=4))
+    ans = input("\n      确认？(Y/n): ").strip().lower()
+    if ans in ("", "y", "yes"):
+        with open("config.json", "w", encoding="utf-8") as f:
+            json.dump(config, f, ensure_ascii=False, indent=4)
+        print("      config.json 已写入")
+    else:
+        print("      已取消，config.json 未修改")
+
+    print("\n配置完成！下一步:")
+    print("  python main.py status    — 查看状态")
+    print("  python main.py decrypt   — 解密数据库")
+    print("  python main.py export    — 解密 + 导出聊天记录")
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="WeChat Decrypt — 配置向导",
+    )
+    parser.add_argument(
+        "--check",
+        action="store_true",
+        help="仅检查环境，不修改文件",
+    )
+    args = parser.parse_args()
+
+    if args.check:
+        check_environment()
+    else:
+        interactive_setup()
+
+
+if __name__ == "__main__":
+    main()