make2/factory_sliceing/build_exe.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
打包脚本：将 factory_sliceing 项目打包成 exe 文件
使用方法: python build_exe.py
"""

import os
import sys
import subprocess
import shutil

def build_exe():
    """使用 PyInstaller 打包项目"""
    
    # 获取当前脚本所在目录（factory_sliceing 目录）
    current_dir = os.path.dirname(os.path.abspath(__file__))
    project_root = os.path.dirname(current_dir)  # make2 目录
    
    # 切换到 factory_sliceing 目录
    os.chdir(current_dir)
    
    # PyInstaller 命令参数
    main_script = os.path.join(current_dir, 'main.py')
    
    # 构建 PyInstaller 命令
    cmd = [
        'pyinstaller',
        '--name=factory_sliceing',  # 生成的 exe 名称
        '--onefile',  # 打包成单个 exe 文件
        '--console',  # 显示控制台窗口（因为需要命令行参数）
        '--clean',  # 清理临时文件
        '--noconfirm',  # 不询问确认
        
        # 添加隐藏导入（如果 PyInstaller 无法自动检测）
        '--hidden-import=auto_sliceing_operate',
        '--hidden-import=auto_sliceing_operate.main_begin_sliceing',
        '--hidden-import=auto_sliceing_operate.main_download_zip',
        '--hidden-import=download_batch_data',
        '--hidden-import=download_batch_data.main_download_batch_data_and_trans',
        '--hidden-import=auto_sliceing_operate.utils',
        '--hidden-import=download_batch_data.utils',
        
        # UI 自动化相关模块
        '--hidden-import=uiautomation',
        
        # Windows API 相关模块（pywin32）
        '--hidden-import=win32gui',
        '--hidden-import=win32con',
        '--hidden-import=win32api',
        '--hidden-import=win32process',
        
        # 其他可能需要的模块
        '--hidden-import=psutil',  # 进程管理（可选，但有 try-except）
        '--hidden-import=minio',  # MinIO 客户端
        '--hidden-import=minio.error',
        
        # 收集 uiautomation 的所有子模块（确保完整打包）
        '--collect-all=uiautomation',
        
        # 添加数据文件（如果需要）
        # '--add-data=config;config',  # Windows 格式
        # '--add-data=config:config',  # Linux/Mac 格式
        
        # 工作目录设置
        f'--workpath={os.path.join(current_dir, "build")}',
        f'--distpath={os.path.join(current_dir, "dist")}',
        f'--specpath={os.path.join(current_dir, "build")}',
        
        main_script
    ]
    
    print("=" * 60)
    print("开始打包 factory_sliceing 项目...")
    print("=" * 60)
    print(f"工作目录: {current_dir}")
    print(f"主脚本: {main_script}")
    print()
    
    try:
        # 检查 PyInstaller 是否安装
        result = subprocess.run(['pyinstaller', '--version'], 
                              capture_output=True, text=True)
        if result.returncode != 0:
            print("错误: PyInstaller 未安装")
            print("请运行: pip install pyinstaller")
            return False
        
        print(f"PyInstaller 版本: {result.stdout.strip()}")
        print()
        
        # 执行打包命令
        print("执行打包命令...")
        print(" ".join(cmd))
        print()
        
        result = subprocess.run(cmd, check=True)
        
        if result.returncode == 0:
            print()
            print("=" * 60)
            print("打包成功！")
            print("=" * 60)
            exe_path = os.path.join(current_dir, 'dist', 'factory_sliceing.exe')
            if os.path.exists(exe_path):
                file_size = os.path.getsize(exe_path) / (1024 * 1024)  # MB
                print(f"生成的 exe 文件: {exe_path}")
                print(f"文件大小: {file_size:.2f} MB")
                print()
                print("使用方法:")
                print("  factory_sliceing.exe <command> <work_dir>")
                print("  可用命令: batch_dwonload | begin_sliceing | download_zip")
            return True
        else:
            print("打包失败！")
            return False
            
    except subprocess.CalledProcessError as e:
        print(f"打包过程中出现错误: {e}")
        return False
    except FileNotFoundError:
        print("错误: 找不到 PyInstaller")
        print("请先安装: pip install pyinstaller")
        return False
    except Exception as e:
        print(f"发生未知错误: {e}")
        return False

def clean_build_files():
    """清理构建产生的临时文件"""
    current_dir = os.path.dirname(os.path.abspath(__file__))
    
    dirs_to_remove = [
        os.path.join(current_dir, 'build'),
        os.path.join(current_dir, '__pycache__'),
    ]
    
    files_to_remove = [
        os.path.join(current_dir, 'factory_sliceing.spec'),
    ]
    
    print("清理临时文件...")
    for dir_path in dirs_to_remove:
        if os.path.exists(dir_path):
            try:
                shutil.rmtree(dir_path)
                print(f"已删除: {dir_path}")
            except Exception as e:
                print(f"删除 {dir_path} 失败: {e}")
    
    for file_path in files_to_remove:
        if os.path.exists(file_path):
            try:
                os.remove(file_path)
                print(f"已删除: {file_path}")
            except Exception as e:
                print(f"删除 {file_path} 失败: {e}")

if __name__ == '__main__':
    import argparse
    
    parser = argparse.ArgumentParser(description='打包 factory_sliceing 项目为 exe')
    parser.add_argument('--clean', action='store_true', 
                       help='打包后清理临时文件')
    parser.add_argument('--clean-only', action='store_true',
                       help='仅清理临时文件，不打包')
    
    args = parser.parse_args()
    
    if args.clean_only:
        clean_build_files()
    else:
        success = build_exe()
        if success and args.clean:
            print()
            clean_build_files()