build.py

"""PatchTool 打包脚本 - 将项目打包为单文件 exe"""

import os
import shutil
import subprocess
import sys

# 项目根目录
ROOT = os.path.dirname(os.path.abspath(__file__))
DIST_DIR = os.path.join(ROOT, "dist")
BUILD_DIR = os.path.join(ROOT, "build_temp")
SPEC_FILE = os.path.join(ROOT, "build.spec")
OUTPUT_EXE = os.path.join(DIST_DIR, "PatchTool.exe")


def check_pyinstaller():
    try:
        import PyInstaller
        print(f"PyInstaller {PyInstaller.__version__} 已就绪")
        return True
    except ImportError:
        print("未安装 PyInstaller，正在安装...")
        subprocess.check_call([sys.executable, "-m", "pip", "install", "pyinstaller"])
        print("PyInstaller 安装完成")
        return True


def clean():
    for d in [BUILD_DIR, DIST_DIR]:
        if os.path.exists(d):
            shutil.rmtree(d)
            print(f"已清理: {d}")


def build():
    cmd = [
        sys.executable, "-m", "PyInstaller",
        SPEC_FILE,
        "--distpath", DIST_DIR,
        "--workpath", BUILD_DIR,
        "--clean", "-y",
    ]
    print(f"执行: {' '.join(cmd)}\n")
    subprocess.check_call(cmd, cwd=ROOT)


def main():
    print("=" * 50)
    print("  PatchTool 打包脚本")
    print("=" * 50)

    check_pyinstaller()
    clean()
    build()

    if os.path.isfile(OUTPUT_EXE):
        size_mb = os.path.getsize(OUTPUT_EXE) / (1024 * 1024)
        print("\n" + "=" * 50)
        print(f"  打包成功!")
        print(f"  输出: {OUTPUT_EXE}")
        print(f"  大小: {size_mb:.1f} MB")
        print("=" * 50)
    else:
        print("\n打包失败，未找到输出文件")
        sys.exit(1)


if __name__ == "__main__":
    main()