BodyBalanceEvaluation/backend/build_app.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
main.py 完整版打包脚本
使用PyInstaller将完整版Flask应用程序（包含SocketIO）打包成独立的exe文件
"""

import os
import sys
import shutil
import subprocess
from pathlib import Path

def clean_build_dirs():
    """清理构建目录"""
    print("清理构建目录...")
    dirs_to_clean = ['build', 'dist', '__pycache__']
    
    for dir_name in dirs_to_clean:
        if os.path.exists(dir_name):
            try:
                shutil.rmtree(dir_name)
                print(f"✓ 已删除 {dir_name}")
            except Exception as e:
                print(f"⚠️  删除 {dir_name} 失败: {e}")

def create_app_spec():
    """创建main.py的PyInstaller spec文件"""
    spec_content = '''
# -*- mode: python ; coding: utf-8 -*-

block_cipher = None

a = Analysis(
    ['main.py'],
    pathex=[],
    binaries=[
        ('dll/femtobolt/bin/k4a.dll', 'dll/femtobolt/bin'),  # K4A动态库
        ('dll/femtobolt/bin/k4arecord.dll', 'dll/femtobolt/bin'),  # K4A录制库
        ('dll/femtobolt/bin/depthengine_2_0.dll', 'dll/femtobolt/bin'),  # 深度引擎
        ('dll/femtobolt/bin/OrbbecSDK.dll', 'dll/femtobolt/bin'),  # Orbbec SDK
        ('dll/femtobolt/bin/ob_usb.dll', 'dll/femtobolt/bin'),  # Orbbec USB库
        ('dll/femtobolt/bin/live555.dll', 'dll/femtobolt/bin'),  # Live555库
        ('dll/femtobolt/bin/OrbbecSDKConfig_v1.0.xml', 'dll/femtobolt/bin'),  # Orbbec配置文件        ('dll/smitsense/SMiTSenseUsb-F3.0.dll', 'dll/smitsense'),  # SMiTSense传感器库
        ('dll/smitsense/SMiTSenseUsb-F3.0.dll', 'dll/smitsense'),  # SMiTSenseUsb库
        ('dll/smitsense/Wrapper.dll', 'dll/smitsense'),  # SMiTSense传感器库包装类
    ],
    hiddenimports=[
        'flask',
        'flask_socketio',
        'flask_cors',
        'cv2',
        'numpy',
        'pandas',
        'scipy',
        'matplotlib',
        'seaborn',
        'sklearn',
        'PIL',
        'reportlab',
        'sqlite3',
        'configparser',
        'logging',
        'threading',
        'queue',
        'base64',
        'psutil',
        'pykinect_azure',
        'pyserial',
        'requests',
        'yaml',
        'click',
        'colorama',
        'tqdm',
        'database',
        'device_manager',
        'utils',
        'eventlet',
        'socketio',
        'engineio',
        'engineio.async_drivers.threading',
        'engineio.async_drivers.eventlet',
        'engineio.async_eventlet',
        'socketio.async_eventlet',
        'greenlet',
        'gevent',
        'gevent.socket',
        'gevent.select',
        'dns',
        'dns.resolver',
        'dns.reversename',
        'dns.e164',
    ],
    hookspath=[],
    hooksconfig={},
    runtime_hooks=[],
    excludes=[],
    win_no_prefer_redirects=False,
    win_private_assemblies=False,
    cipher=block_cipher,
    noarchive=False,
)

pyz = PYZ(a.pure, a.zipped_data, cipher=block_cipher)

exe = EXE(
    pyz,
    a.scripts,
    [],
    exclude_binaries=True,
    name='BodyBalanceBackend',
    debug=False,
    bootloader_ignore_signals=False,
    strip=False,
    upx=True,
    upx_exclude=[],
    runtime_tmpdir=None,
    console=True,
    disable_windowed_traceback=False,
    argv_emulation=False,
    target_arch=None,
    codesign_identity=None,
    entitlements_file=None,
    icon=None
)

coll = COLLECT(
    exe,
    a.binaries,
    a.zipfiles,
    a.datas,
    strip=False,
    upx=True,
    upx_exclude=[],
    name='BodyBalanceBackend'
)
'''
    
    with open('app.spec', 'w', encoding='utf-8') as f:
        f.write(spec_content)
    
    print("✓ 已创建 app.spec 文件")

def build_exe():
    """构建exe文件"""
    print("\n开始构建exe文件...")
    
    try:
        # 使用PyInstaller构建
        cmd = [sys.executable, '-m', 'PyInstaller', 'app.spec', '--clean', '--noconfirm']
        print(f"执行命令: {' '.join(cmd)}")
        
        # 修改编码处理，避免UTF-8错误
        result = subprocess.run(cmd, capture_output=True, text=True, encoding='utf-8', errors='ignore')
        
        if result.returncode == 0:
            print("✓ 构建成功！")
            return True
        else:
            print(f"✗ 构建失败")
            print(f"错误输出: {result.stderr}")
            return False
            
    except Exception as e:
        print(f"✗ 构建过程出错: {e}")
        return False


def create_directories():
    """创建必要的目录结构"""
    print("创建目录结构...")
    
    directories = [
        os.path.join('dist', 'BodyBalanceBackend', 'dll'),
        os.path.join('dist', 'BodyBalanceBackend', 'data'),
        os.path.join('dist', 'BodyBalanceBackend', 'logs')
    ]
    
    for directory in directories:
        if not os.path.exists(directory):
            os.makedirs(directory)
            print(f"✓ 创建目录: {os.path.relpath(directory, 'dist')}/")
        else:
            print(f"ℹ️  目录已存在: {os.path.relpath(directory, 'dist')}/")

def copy_dll_files():
    """复制DLL文件到dll目录"""
    print("复制DLL文件...")
    
    dll_dir = os.path.join('dist', 'BodyBalanceBackend', 'dll')
    
    # 查找所有DLL文件
    dll_files = []
    
    # 复制dll目录下的所有文件夹和文件到目标目录的dll目录下
    source_dll_dir = 'dll'
    if not os.path.exists(source_dll_dir):
        print("⚠️  未找到dll目录")
        return
    
    try:
        # 遍历dll目录下的所有内容
        for item in os.listdir(source_dll_dir):
            source_path = os.path.join(source_dll_dir, item)
            target_path = os.path.join(dll_dir, item)
            
            if os.path.isdir(source_path):
                # 如果是目录，递归复制整个目录
                shutil.copytree(source_path, target_path, dirs_exist_ok=True)
                print(f"✓ 已复制目录 {item}")
            else:
                # 如果是文件，直接复制
                shutil.copy2(source_path, target_path)
                print(f"✓ 已复制文件 {item}")
    except Exception as e:
        print(f"⚠️  复制dll目录内容失败: {e}")

def copy_data_files():
    """准备数据库文件到data目录"""
    print("准备数据库文件...")
    
    # 数据库文件应该复制到可执行文件同级的data目录
    data_dir = os.path.join('dist', 'BodyBalanceBackend', 'data')
    
    # 数据库文件列表 - 这些文件会在程序首次运行时自动创建
    db_files = ['body_balance.db']
    
    # 检查是否存在现有数据库文件，如果存在则复制
    copied_count = 0
    for db_file in db_files:
        # 检查当前目录和data目录中的数据库文件
        source_paths = [db_file, os.path.join('data', db_file)]
        
        for source_path in source_paths:
            if os.path.exists(source_path):
                try:
                    shutil.copy2(source_path, data_dir)
                    print(f"✓ 已复制 {source_path} -> {data_dir}")
                    copied_count += 1
                    break  # 找到一个就跳出内层循环
                except Exception as e:
                    print(f"⚠️  复制 {source_path} 失败: {e}")
    
    if copied_count == 0:
        print("ℹ️  未找到现有数据库文件，程序首次运行时将自动创建")
    else:
        print(f"✓ 成功复制 {copied_count} 个数据库文件")

def copy_config_files():
    """复制配置文件到dist目录"""
    print("复制配置文件...")
    
    config_files = ['config.ini']
    dist_dir = os.path.join('dist', 'BodyBalanceBackend')
    
    if not os.path.exists(dist_dir):
        os.makedirs(dist_dir)
    
    for config_file in config_files:
        if os.path.exists(config_file):
            try:
                shutil.copy2(config_file, dist_dir)
                print(f"✓ 已复制 {config_file}")
            except Exception as e:
                print(f"⚠️  复制 {config_file} 失败: {e}")
        else:
            print(f"⚠️  配置文件不存在: {config_file}")

def main():
    """主函数"""
    print("=" * 60)
    print("身体平衡评估系统 - main.py 完整版打包工具")
    print("=" * 60)
    print()
    
    # 检查当前目录
    if not os.path.exists('main.py'):
        print("✗ 错误: 找不到 main.py 文件")
        print("请确保在backend目录下运行此脚本")
        input("按回车键退出...")
        return
    
    try:
        # 清理构建目录
        clean_build_dirs()
        print()
        
        # 创建spec文件
        print("创建PyInstaller配置...")
        create_app_spec()
        print()
        
        # 构建exe
        if build_exe():
            print()
            print("后处理...")
            
            # 检查生成的exe文件
            exe_path = 'dist/BodyBalanceBackend/BodyBalanceBackend.exe'
            if os.path.exists(exe_path):
                print(f"✓ exe文件位置: {exe_path}")
                
                # 创建目录结构
                create_directories()
                print()
                
                # 复制DLL文件
                copy_dll_files()
                print()
                
                # 复制数据库文件
                copy_data_files()
                print()
                
                # 复制配置文件
                copy_config_files()
                print()
                


                
                print()
                print("🎉 打包完成！")
                print()
                print("输出文件:")
                print(f"- 可执行文件: {exe_path}")
                print("- 启动脚本: dist/start_backend.bat")
                print("- 配置文件: dist/config.ini")
                print()
                print("目录结构:")
                print("- BodyBalanceBackend/ - 应用程序文件夹")
                print("- dll/               - DLL文件")
                print("- data/              - 数据库文件")
                print("- logs/              - 日志文件")
                print()
                print("使用方式:")
                print("1. 直接运行: dist/BodyBalanceBackend/BodyBalanceBackend.exe")
                print("2. 使用脚本: dist/start_backend.bat")
                print()
                print("服务地址: http://localhost:5000")
                print("SocketIO: ws://localhost:5000")
                
            else:
                print("✗ 错误: 未找到生成的exe文件")
        else:
            print("\n✗ 打包失败")
    
    except Exception as e:
        print(f"\n✗ 打包过程出错: {e}")
    
    print()
    input("按回车键退出...")

if __name__ == '__main__':
    main()