import os
import subprocess
import json
import glob
import platform

# === 配置区域 ===
OUTPUT_FILE = "requirements.txt"
# 忽略的基础工具包
IGNORE_PKGS = {
    "pip", "setuptools", "wheel", "distribute", "wsgiref", "argparse", "python-dateutil"
}

def find_python_interpreters():
    """自动寻找系统中所有可能的 Python 解释器路径"""
    python_paths = set()

    # 1. 检查环境变量中的 Python
    try:
        if platform.system() == "Windows":
            cmd = "where python"
        else:
            cmd = "which -a python3 python"
        
        paths = subprocess.check_output(cmd, shell=True, text=True).strip().split('\n')
        for p in paths:
            if p and os.path.isfile(p):
                python_paths.add(p.strip())
    except:
        pass

    # 2. 检查 Python Launcher 记录的路径
    if platform.system() == "Windows":
        try:
            py_out = subprocess.check_output("py --list-paths", shell=True, text=True)
            for line in py_out.splitlines():
                if "python.exe" in line:
                    # 提取路径
                    path = line.split()[-1]
                    python_paths.add(path)
        except:
            pass

        # 3. 暴力扫描常见安装目录
        user_profile = os.environ.get("USERPROFILE", "")
        search_patterns = [
            r"C:\Python*\python.exe",  # C盘根目录旧版本
            r"C:\Program Files\Python*\python.exe",
            r"C:\Program Files (x86)\Python*\python.exe",
            os.path.join(user_profile, r"AppData\Local\Programs\Python\Python*\python.exe"), # 默认安装位
            os.path.join(user_profile, r"scoop\apps\python\*\python.exe"), # Scoop
        ]

        for pattern in search_patterns:
            for p in glob.glob(pattern):
                python_paths.add(p)

    return sorted(list(python_paths))

def get_packages_from_python(python_path):
    """提取指定 Python 的包列表"""
    print(f"📦 正在扫描: {python_path} ...")
    try:
        # 使用 json 格式获取
        cmd = [python_path, "-m", "pip", "list", "--format=json", "--disable-pip-version-check"]
        result = subprocess.run(cmd, capture_output=True, text=True)

        if result.returncode != 0:
            print(f"   ⚠️ 无法读取 (可能未安装pip): {python_path}")
            return set()

        data = json.loads(result.stdout)
        pkgs = set()
        for item in data:
            name = item['name']
            if name.lower() not in IGNORE_PKGS:
                pkgs.add(name) # 只存名字，不存版本号
        print(f"   ✅ 发现 {len(pkgs)} 个包")
        return pkgs
    except Exception as e:
        print(f"   ❌ 出错: {e}")
        return set()

def main():
    interpreters = find_python_interpreters()
    print(f"\n🔎 找到了 {len(interpreters)} 个 Python 解释器。开始提取依赖...\n")

    all_packages = set()

    for py_path in interpreters:
        pkgs = get_packages_from_python(py_path)
        all_packages.update(pkgs)

    # 保存结果
    with open(OUTPUT_FILE, "w", encoding="utf-8") as f:
        for pkg in sorted(all_packages):
            f.write(f"{pkg}\n")

    print(f"\n🎉 共合并了 {len(all_packages)} 个唯一的库，清单已保存至当前目录的: {OUTPUT_FILE}")

if __name__ == "__main__":
    main()