78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
import os
|
||
import shutil
|
||
import subprocess
|
||
import sys
|
||
|
||
def main():
|
||
# 1. 定义路径
|
||
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
||
DIST_DIR = os.path.join(BASE_DIR, 'dist')
|
||
APP_NAME = 'Depression_Decoder'
|
||
TARGET_DIR = os.path.join(DIST_DIR, APP_NAME)
|
||
|
||
MODEL_SRC = os.path.join(BASE_DIR, 'model')
|
||
RAW_DATA_SRC = os.path.join(BASE_DIR, 'raw_data')
|
||
|
||
MODEL_DST = os.path.join(TARGET_DIR, 'model')
|
||
RAW_DATA_DST = os.path.join(TARGET_DIR, 'raw_data')
|
||
|
||
# 2. 清理旧构建
|
||
print("[1/3] Cleaning up old builds...")
|
||
if os.path.exists(DIST_DIR):
|
||
try:
|
||
shutil.rmtree(DIST_DIR)
|
||
print(" Cleaned dist/")
|
||
except Exception as e:
|
||
print(f" Warning: Could not clean dist/: {e}")
|
||
|
||
BUILD_DIR = os.path.join(BASE_DIR, 'build')
|
||
if os.path.exists(BUILD_DIR):
|
||
try:
|
||
shutil.rmtree(BUILD_DIR)
|
||
print(" Cleaned build/")
|
||
except Exception as e:
|
||
print(f" Warning: Could not clean build/: {e}")
|
||
|
||
# 3. 运行 PyInstaller
|
||
print("[2/3] Running PyInstaller...")
|
||
# 注意：我们这里不传 --noupx，因为已经在 spec 文件里把 upx=False 写死了
|
||
cmd = [
|
||
"pyinstaller",
|
||
"build_algorithm.spec",
|
||
"--clean"
|
||
]
|
||
|
||
try:
|
||
subprocess.check_call(cmd, shell=True)
|
||
except subprocess.CalledProcessError:
|
||
print("Error: PyInstaller failed.")
|
||
sys.exit(1)
|
||
|
||
# 4. 复制外部资源文件夹
|
||
print("[3/3] Copying external resources...")
|
||
|
||
# 复制 model 文件夹
|
||
if os.path.exists(MODEL_SRC):
|
||
if os.path.exists(MODEL_DST):
|
||
shutil.rmtree(MODEL_DST)
|
||
shutil.copytree(MODEL_SRC, MODEL_DST)
|
||
print(f" Copied: model -> {MODEL_DST}")
|
||
else:
|
||
print(f" Warning: Source model dir not found at {MODEL_SRC}")
|
||
|
||
# 复制 raw_data 文件夹
|
||
if os.path.exists(RAW_DATA_SRC):
|
||
if os.path.exists(RAW_DATA_DST):
|
||
shutil.rmtree(RAW_DATA_DST)
|
||
shutil.copytree(RAW_DATA_SRC, RAW_DATA_DST)
|
||
print(f" Copied: raw_data -> {RAW_DATA_DST}")
|
||
else:
|
||
print(f" Warning: Source raw_data dir not found at {RAW_DATA_SRC}")
|
||
|
||
print("\n" + "="*50)
|
||
print(f"SUCCESS! Build artifacts are in: {TARGET_DIR}")
|
||
print("="*50)
|
||
|
||
if __name__ == "__main__":
|
||
main()
|