88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
import os
|
||
import subprocess
|
||
import sys
|
||
import shutil
|
||
|
||
# 确保我们在虚拟环境中运行
|
||
if not sys.prefix == sys.base_prefix:
|
||
print(f"正在使用虚拟环境: {sys.prefix}")
|
||
else:
|
||
print("警告：你似乎没有激活虚拟环境！建议在 venv_clean 下运行。")
|
||
|
||
|
||
def build():
|
||
entry_point = "runDecoder.py" # 你的入口文件
|
||
|
||
# 自动清理逻辑优化
|
||
output_dir = "dist2"
|
||
build_dir = "build2" # Nuitka 默认会在当前目录生成 .build 文件夹
|
||
|
||
if "--clean" in sys.argv:
|
||
print("清理旧构建目录...")
|
||
for folder in [output_dir, build_dir, entry_point.replace(".py", ".build")]:
|
||
if os.path.exists(folder):
|
||
shutil.rmtree(folder, ignore_errors=True)
|
||
|
||
# Nuitka 命令 - 此时非常清爽
|
||
nuitka_cmd = [
|
||
sys.executable, "-m", "nuitka",
|
||
"--standalone", # 独立运行模式
|
||
f"--output-dir={output_dir}", # 输出目录
|
||
"--show-progress", # 显示进度
|
||
"--assume-yes-for-downloads", # 自动下载依赖（如 ccache, depends 等）
|
||
|
||
# --- 插件配置 ---
|
||
"--enable-plugin=numpy",
|
||
"--enable-plugin=matplotlib",
|
||
"--enable-plugin=torch", # 处理 PyTorch 及其 CUDA 依赖
|
||
|
||
# --- 包含包/模块 (Nuitka 2.x 推荐使用 include-package-data 或 include-package) ---
|
||
# --collect-all 是 PyInstaller 的参数，Nuitka 不支持
|
||
"--include-package=sklearn",
|
||
"--include-package=scipy",
|
||
"--include-package=mne",
|
||
|
||
# 强制包含 MNE 的数据文件（配置、布局等）
|
||
"--include-package-data=mne",
|
||
"--include-package=PIL", # Pillow (matplotlib/mne 可能用到)
|
||
"--include-package=networkx", # mne 可能用到
|
||
"--include-package=decorator", # MNE 核心依赖，防止 KeyError: 'self'
|
||
"--include-package=six", # 通用兼容库
|
||
|
||
# 显式包含本地模块，防止隐式导入丢失
|
||
"--include-module=infer_pth",
|
||
|
||
# --- 数据文件 ---
|
||
# 格式: 源路径=目标路径 (相对 dist 目录)
|
||
"--include-data-dir=model=model",
|
||
"--include-data-dir=raw_data=raw_data",
|
||
|
||
# --- 排除干扰以减小体积/提高稳定性 ---
|
||
"--nofollow-import-to=pytest",
|
||
"--nofollow-import-to=unittest",
|
||
"--nofollow-import-to=pdb",
|
||
"--nofollow-import-to=tkinter", # 如果不用 GUI 界面
|
||
"--nofollow-import-to=sympy", # 除非明确用到符号计算
|
||
|
||
# --- 内存与性能 ---
|
||
"--low-memory", # 降低打包时的内存消耗
|
||
|
||
# --- Windows 特定 ---
|
||
# "--disable-console", # 如果不需要黑框，取消注释这一行
|
||
]
|
||
|
||
nuitka_cmd.append(entry_point)
|
||
|
||
print("开始打包...")
|
||
try:
|
||
subprocess.check_call(nuitka_cmd)
|
||
print("\n打包成功！")
|
||
print(f"请在 dist2/runDecoder.dist 目录下运行 exe 进行测试。")
|
||
|
||
except subprocess.CalledProcessError as e:
|
||
print(f"打包失败，错误码: {e.returncode}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
build()
|