230 lines
7.5 KiB
Python
230 lines
7.5 KiB
Python
from __future__ import annotations
|
||
|
||
import os
|
||
import shutil
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
|
||
PROJECT_ROOT = Path(__file__).resolve().parents[2]
|
||
TOOLS_DIR = Path(__file__).resolve().parent
|
||
|
||
|
||
def run(cmd: list[str], cwd: Path | None = None) -> str:
|
||
proc = subprocess.run(cmd, cwd=str(cwd or PROJECT_ROOT), capture_output=True, text=True, shell=False)
|
||
if proc.returncode != 0:
|
||
raise RuntimeError(f"Command failed: {' '.join(cmd)}\nstdout:\n{proc.stdout}\nstderr:\n{proc.stderr}")
|
||
return proc.stdout.strip()
|
||
|
||
|
||
def get_current_tag() -> str:
|
||
# 优先使用当前 tag；若没有打 tag，则用短 commit id
|
||
try:
|
||
tag = run(["git", "describe", "--tags", "--exact-match"]).strip()
|
||
if tag:
|
||
return tag
|
||
except Exception:
|
||
pass
|
||
# 回退到当前 commit 短哈希
|
||
try:
|
||
short = run(["git", "rev-parse", "--short", "HEAD"]).strip()
|
||
return f"commit-{short}"
|
||
except Exception:
|
||
return "untagged"
|
||
|
||
|
||
def read_git_ignored_paths() -> set[str]:
|
||
"""
|
||
读取 .gitignore（若存在）并返回需忽略的模式集合（简单前缀/目录名过滤）。
|
||
我们只用于资源复制时的粗过滤；最终以 PyInstaller 的 --exclude/--add-data 控制。
|
||
"""
|
||
ignored: set[str] = set()
|
||
gi = PROJECT_ROOT / ".gitignore"
|
||
if not gi.exists():
|
||
return ignored
|
||
for line in gi.read_text(encoding="utf-8").splitlines():
|
||
s = line.strip()
|
||
if not s or s.startswith("#"):
|
||
continue
|
||
ignored.add(s)
|
||
return ignored
|
||
|
||
|
||
def is_path_ignored(path: Path, ignored_patterns: set[str]) -> bool:
|
||
# 仅用简单规则过滤常见目录：logs、台本、cache、__pycache__、*.log、TODO、*.md 临时内容等
|
||
name = path.name
|
||
rel = path.relative_to(PROJECT_ROOT).as_posix()
|
||
if name in {"logs", "台本", "cache", "__pycache__"}:
|
||
return True
|
||
if name.lower() in {"todo"}:
|
||
return True
|
||
if rel.startswith("tools/package/"):
|
||
return True
|
||
if any(seg == "__pycache__" for seg in path.parts):
|
||
return True
|
||
if path.suffix.lower() in {".log"}:
|
||
return True
|
||
# 粗略匹配 .gitignore 的以目录结尾的规则
|
||
for pat in ignored_patterns:
|
||
if pat.endswith("/") and rel.startswith(pat.rstrip("/")):
|
||
return True
|
||
if pat == rel or rel.startswith(pat.rstrip("/")):
|
||
# 简化：若规则与开头匹配，则跳过
|
||
return True
|
||
return False
|
||
|
||
|
||
def build_with_pyinstaller(output_dir: Path) -> None:
|
||
# 入口脚本
|
||
entry = PROJECT_ROOT / "src" / "run" / "run.py"
|
||
|
||
# 资源目录：assets 与 static（排除 static/local_config.yml）
|
||
add_data_args: list[str] = []
|
||
assets_dir = PROJECT_ROOT / "assets"
|
||
static_dir = PROJECT_ROOT / "static"
|
||
tmp_dir = output_dir / "_tmp_resources"
|
||
tmp_dir.mkdir(parents=True, exist_ok=True)
|
||
if assets_dir.exists():
|
||
add_data_args += ["--add-data", f"{assets_dir}{os.pathsep}assets"]
|
||
if static_dir.exists():
|
||
# 构建一个不包含 local_config.yml 的临时 static 目录
|
||
tmp_static = tmp_dir / "static"
|
||
if tmp_static.exists():
|
||
shutil.rmtree(tmp_static)
|
||
shutil.copytree(static_dir, tmp_static)
|
||
lc = tmp_static / "local_config.yml"
|
||
if lc.exists():
|
||
lc.unlink()
|
||
add_data_args += ["--add-data", f"{tmp_static}{os.pathsep}static"]
|
||
|
||
# 额外的排除（减少包体）
|
||
exclude_modules = [
|
||
"tests",
|
||
"unittest",
|
||
"tkinter",
|
||
"pytest",
|
||
"matplotlib",
|
||
]
|
||
exclude_args: list[str] = []
|
||
for m in exclude_modules:
|
||
exclude_args += ["--exclude-module", m]
|
||
|
||
# 运行 PyInstaller（优先单目录，兼容资源文件；不开启 --onefile 以避免 pygame 资源路径问题）
|
||
dist_dir = output_dir
|
||
build_dir = output_dir / "build"
|
||
spec_path = output_dir
|
||
build_dir.mkdir(parents=True, exist_ok=True)
|
||
dist_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
cmd = [
|
||
sys.executable, "-m", "PyInstaller",
|
||
"--noconfirm",
|
||
"--clean",
|
||
"--name", "cultivation-world-simulator",
|
||
"--distpath", str(dist_dir),
|
||
"--workpath", str(build_dir),
|
||
"--specpath", str(spec_path),
|
||
"--console",
|
||
# 去掉调试与符号，减小体积
|
||
"--optimize", "2",
|
||
# 隐式集合数据（若依赖包有数据）
|
||
"--collect-all", "omegaconf",
|
||
"--collect-all", "json5",
|
||
"--collect-submodules", "pygame",
|
||
] + exclude_args + add_data_args + [str(entry)]
|
||
|
||
print("[1/3] 调用 PyInstaller...")
|
||
print("命令:", " ".join(cmd))
|
||
run(cmd)
|
||
print("PyInstaller 完成。")
|
||
|
||
|
||
def copy_project_side_files(output_dir: Path, tag_name: str) -> None:
|
||
print("[2/3] 复制说明与许可证...")
|
||
app_dir = output_dir / "cultivation-world-simulator"
|
||
app_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 将 README、LICENSE、requirements.txt 复制到应用目录，便于分发
|
||
for fname in ["README.md", "EN_README.md", "LICENSE", "requirements.txt"]:
|
||
src = PROJECT_ROOT / fname
|
||
if src.exists():
|
||
dst = app_dir / fname
|
||
shutil.copy2(src, dst)
|
||
|
||
# 生成一个运行说明
|
||
(app_dir / "HOW_TO_RUN.txt").write_text(
|
||
(
|
||
"运行说明:\n"
|
||
"1) 双击 cultivation-world-simulator/cultivation-world-simulator.exe 启动\n"
|
||
"2) 如需配置 LLM，请编辑 static/config.yml 或在外部同目录提供 static/local_config.yml 覆盖\n"
|
||
f"版本: {tag_name}\n"
|
||
),
|
||
encoding="utf-8",
|
||
)
|
||
|
||
|
||
def _copy_env_installer(app_dir: Path) -> None:
|
||
print("[3/3] 放置用户安装脚本...")
|
||
src_cmd = TOOLS_DIR / "set_env.cmd"
|
||
if src_cmd.exists():
|
||
# 复制到 exe 同目录，并重命名
|
||
dst_cmd = app_dir / "启动前点击安装.exe"
|
||
try:
|
||
shutil.copy2(src_cmd, dst_cmd)
|
||
except Exception as e:
|
||
print(f"警告：复制 set_env.cmd 失败：{e}")
|
||
else:
|
||
print("提示：未找到 tools/package/set_env.cmd，跳过复制。")
|
||
|
||
|
||
def main() -> None:
|
||
tag = get_current_tag()
|
||
# 输出改为项目根 tmp/{tag}
|
||
release_dir = PROJECT_ROOT / "tmp" / tag
|
||
|
||
# 清理旧目录
|
||
if release_dir.exists():
|
||
print(f"清理旧目录: {release_dir}")
|
||
shutil.rmtree(release_dir)
|
||
release_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
build_with_pyinstaller(release_dir)
|
||
copy_project_side_files(release_dir, tag)
|
||
_copy_env_installer(release_dir / "cultivation-world-simulator")
|
||
|
||
# 删除多余：构建中间产物
|
||
build_dir = release_dir / "build"
|
||
spec_file = release_dir / "cultivation-world-simulator.spec"
|
||
if build_dir.exists():
|
||
shutil.rmtree(build_dir)
|
||
if spec_file.exists():
|
||
spec_file.unlink()
|
||
# 删除临时资源
|
||
tmp_dir = release_dir / "_tmp_resources"
|
||
if tmp_dir.exists():
|
||
shutil.rmtree(tmp_dir)
|
||
|
||
# 确保没有把 gitignore 指定的目录带入（单目录输出仅包含 PyInstaller 产物与我们复制的文件）
|
||
ignored = read_git_ignored_paths()
|
||
for path in release_dir.rglob("*"):
|
||
if is_path_ignored(path, ignored):
|
||
if path.is_file():
|
||
try:
|
||
path.unlink()
|
||
except Exception:
|
||
pass
|
||
else:
|
||
try:
|
||
shutil.rmtree(path)
|
||
except Exception:
|
||
pass
|
||
|
||
print(f"打包完成: {release_dir}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|
||
|
||
|