build_pyinstaller.py

# -*- coding: utf-8 -*-
"""
PyBuilder - PyInstaller 构建脚本
版本: 1.0.0
"""

import sys
import os
import subprocess
import shutil
import time
import platform


# ANSI 颜色代码
class Color:
    RESET = "\033[0m"
    BOLD = "\033[1m"
    GREEN = "\033[92m"
    YELLOW = "\033[93m"
    RED = "\033[91m"
    CYAN = "\033[96m"
    GRAY = "\033[90m"


# 构建配置
PROJECT_NAME = "PyBuilder"
VERSION = "1.0.0"
COMPANY_NAME = "ASLant"
ENTRY_FILE = "main.py"
ICON_FILE = "assets/app.ico"
OUTPUT_DIR = "build"


def build():
    """执行 PyInstaller 构建"""
    # 获取平台信息
    os_type = platform.system()
    is_windows = os_type == "Windows"
    is_macos = os_type == "Darwin"

    # 获取终端宽度
    width = shutil.get_terminal_size().columns
    separator = "-" * width
    start_time = time.time()

    print(
        f"{Color.CYAN}{Color.BOLD}Building {PROJECT_NAME} v{VERSION} on {os_type}{Color.RESET}"
    )
    print(separator)

    # 添加数据文件（根据操作系统使用不同分隔符）
    data_separator = ";" if is_windows else ":"

    # 构建 PyInstaller 命令
    cmd = [
        sys.executable,
        "-m",
        "PyInstaller",
        f"--distpath={OUTPUT_DIR}",
        "--workpath=build/temp",
        f"--name={PROJECT_NAME}",
        "--contents-directory=lib",
        "--clean",
        "--noconfirm",
        "--log-level=WARN",
    ]

    # 图标文件（仅Windows和macOS平台）
    if is_windows or is_macos:
        cmd.append(f"--icon={ICON_FILE}")

    # 收集子模块
    cmd.append("--collect-submodules=textual")

    # 添加数据文件
    cmd.append(
        f"--add-data={os.path.join('src', 'style')}{data_separator}{os.path.join('src', 'style')}"
    )
    cmd.append(
        f"--add-data={os.path.join('assets', 'pyfiglet')}{data_separator}{'pyfiglet'}"
    )
    cmd.append(f"--add-data={'docs'}{data_separator}{'docs'}")

    # 添加入口文件
    cmd.append(ENTRY_FILE)

    # 执行构建
    print(f"{Color.GRAY}Command:{Color.RESET}")
    print(f"{Color.GRAY}" + " ".join(cmd) + f"{Color.RESET}")
    print(separator)
    print(f"{Color.YELLOW}Building, please wait...{Color.RESET}")
    print()

    try:
        subprocess.run(cmd, check=True)
        print(separator)
        elapsed_time = time.time() - start_time
        minutes = int(elapsed_time // 60)
        seconds = int(elapsed_time % 60)
        print(f"{Color.GREEN}{Color.BOLD}Build successful!{Color.RESET}")
        # 清理 .spec 文件
        spec_file = f"{PROJECT_NAME}.spec"
        if os.path.exists(spec_file):
            os.remove(spec_file)
            print(f"{Color.GRAY}Cleaned: {spec_file}{Color.RESET}")
        abs_output = os.path.abspath(OUTPUT_DIR)
        print(f"{Color.GREEN}Output: {abs_output}{Color.RESET}")
        if minutes > 0:
            print(f"{Color.CYAN}Build time: {minutes}m {seconds}s{Color.RESET}")
        else:
            print(f"{Color.CYAN}Build time: {seconds}s{Color.RESET}")
        return 0
    except subprocess.CalledProcessError as e:
        print(separator)
        print(
            f"{Color.RED}{Color.BOLD}Build failed: {Color.RESET}{Color.RED}{e}{Color.RESET}"
        )
        return 1
    except Exception as e:
        print(separator)
        print(f"{Color.RED}{Color.BOLD}Error: {Color.RESET}{Color.RED}{e}{Color.RESET}")
        return 1


if __name__ == "__main__":
    sys.exit(build())