73 lines
2.2 KiB
Python
73 lines
2.2 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'
|
||
|
|
|
||
|
|
MODEL_SRC = os.path.join(BASE_DIR, 'model')
|
||
|
|
RAW_DATA_SRC = os.path.join(BASE_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...")
|
||
|
|
cmd = [
|
||
|
|
"pyinstaller",
|
||
|
|
"build_algorithm.spec",
|
||
|
|
"--clean",
|
||
|
|
"--noconfirm"
|
||
|
|
]
|
||
|
|
|
||
|
|
try:
|
||
|
|
subprocess.check_call(cmd, shell=True)
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
print("Error: PyInstaller failed.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
# 4. 复制外部资源 (如果存在)
|
||
|
|
print("[3/3] Copying external resources...")
|
||
|
|
|
||
|
|
# 确保 dist 目录存在 (pyinstaller 应该已经创建了)
|
||
|
|
if not os.path.exists(DIST_DIR):
|
||
|
|
os.makedirs(DIST_DIR)
|
||
|
|
|
||
|
|
for src_path, folder_name in [(MODEL_SRC, 'model'), (RAW_DATA_SRC, 'raw_data')]:
|
||
|
|
dst_path = os.path.join(DIST_DIR, folder_name)
|
||
|
|
if os.path.exists(src_path):
|
||
|
|
try:
|
||
|
|
if os.path.exists(dst_path):
|
||
|
|
shutil.rmtree(dst_path)
|
||
|
|
shutil.copytree(src_path, dst_path)
|
||
|
|
print(f" Copied {folder_name} to dist/")
|
||
|
|
except Exception as e:
|
||
|
|
print(f" Error copying {folder_name}: {e}")
|
||
|
|
else:
|
||
|
|
print(f" Note: {folder_name} source not found at {src_path}, skipping.")
|
||
|
|
|
||
|
|
print("\n" + "="*50)
|
||
|
|
print(f"SUCCESS! Executable is in: {DIST_DIR}")
|
||
|
|
print("="*50)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|