138 lines
3.9 KiB
Python
138 lines
3.9 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
PyInstaller 打包验证脚本
|
||
|
|
用于在没有 EEG 设备的情况下验证打包是否成功
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import subprocess
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
def check_pyinstaller_installed():
|
||
|
|
"""检查 PyInstaller 是否安装"""
|
||
|
|
try:
|
||
|
|
result = subprocess.run(['pyinstaller', '--version'],
|
||
|
|
capture_output=True, text=True)
|
||
|
|
print(f"✓ PyInstaller 版本: {result.stdout.strip()}")
|
||
|
|
return True
|
||
|
|
except FileNotFoundError:
|
||
|
|
print("✗ PyInstaller 未安装")
|
||
|
|
return False
|
||
|
|
|
||
|
|
def check_dist_folder():
|
||
|
|
"""检查 dist 文件夹是否存在"""
|
||
|
|
base_dir = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
dist_dir = os.path.join(base_dir, 'dist', 'start_parse')
|
||
|
|
|
||
|
|
if os.path.exists(dist_dir):
|
||
|
|
print(f"✓ dist 文件夹存在: {dist_dir}")
|
||
|
|
|
||
|
|
# 检查 exe 文件
|
||
|
|
exe_path = os.path.join(dist_dir, 'start_parse.exe')
|
||
|
|
if os.path.exists(exe_path):
|
||
|
|
size_mb = os.path.getsize(exe_path) / (1024 * 1024)
|
||
|
|
print(f"✓ 可执行文件存在: start_parse.exe ({size_mb:.1f} MB)")
|
||
|
|
else:
|
||
|
|
print("✗ 可执行文件不存在")
|
||
|
|
return False
|
||
|
|
|
||
|
|
# 检查资源文件
|
||
|
|
xlsx_path = os.path.join(dist_dir, 'xy_64.xlsx')
|
||
|
|
if os.path.exists(xlsx_path):
|
||
|
|
print(f"✓ 资源文件存在: xy_64.xlsx")
|
||
|
|
else:
|
||
|
|
print("✗ 资源文件 xy_64.xlsx 不存在")
|
||
|
|
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
print(f"✗ dist 文件夹不存在，请先运行打包")
|
||
|
|
return False
|
||
|
|
|
||
|
|
def check_dependencies():
|
||
|
|
"""检查关键依赖是否在打包中"""
|
||
|
|
base_dir = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
dist_dir = os.path.join(base_dir, 'dist', 'start_parse')
|
||
|
|
|
||
|
|
if not os.path.exists(dist_dir):
|
||
|
|
return False
|
||
|
|
|
||
|
|
# 检查关键 DLL 文件
|
||
|
|
critical_dlls = [
|
||
|
|
# zmq 依赖
|
||
|
|
'libzmq.pyd',
|
||
|
|
# numpy 依赖
|
||
|
|
'numpy.core._multiarray_umath.cp310-win_amd64.pyd',
|
||
|
|
# scipy 依赖
|
||
|
|
'scipy.special._ufuncs.cp310-win_amd64.pyd',
|
||
|
|
]
|
||
|
|
|
||
|
|
print("\n检查关键依赖文件:")
|
||
|
|
found_count = 0
|
||
|
|
for dll in critical_dlls:
|
||
|
|
found = False
|
||
|
|
for root, dirs, files in os.walk(dist_dir):
|
||
|
|
if dll in files:
|
||
|
|
found = True
|
||
|
|
break
|
||
|
|
status = "✓" if found else "✗"
|
||
|
|
print(f" {status} {dll}")
|
||
|
|
if found:
|
||
|
|
found_count += 1
|
||
|
|
|
||
|
|
return found_count >= len(critical_dlls) // 2
|
||
|
|
|
||
|
|
def test_imports():
|
||
|
|
"""测试关键模块是否可以导入"""
|
||
|
|
print("\n测试模块导入:")
|
||
|
|
|
||
|
|
modules = ['zmq', 'serial', 'numpy', 'pandas', 'scipy']
|
||
|
|
success = True
|
||
|
|
|
||
|
|
for mod in modules:
|
||
|
|
try:
|
||
|
|
__import__(mod)
|
||
|
|
print(f" ✓ {mod}")
|
||
|
|
except ImportError as e:
|
||
|
|
print(f" ✗ {mod}: {e}")
|
||
|
|
success = False
|
||
|
|
|
||
|
|
return success
|
||
|
|
|
||
|
|
def main():
|
||
|
|
print("=" * 60)
|
||
|
|
print("PyInstaller 打包验证")
|
||
|
|
print("=" * 60)
|
||
|
|
|
||
|
|
checks = [
|
||
|
|
("1. 检查 PyInstaller 安装", check_pyinstaller_installed),
|
||
|
|
("2. 检查 dist 文件夹", check_dist_folder),
|
||
|
|
("3. 检查依赖文件", check_dependencies),
|
||
|
|
("4. 测试模块导入", test_imports),
|
||
|
|
]
|
||
|
|
|
||
|
|
results = []
|
||
|
|
for name, check_func in checks:
|
||
|
|
print(f"\n{name}")
|
||
|
|
print("-" * 40)
|
||
|
|
results.append(check_func())
|
||
|
|
|
||
|
|
print("\n" + "=" * 60)
|
||
|
|
print("验证结果汇总:")
|
||
|
|
print("=" * 60)
|
||
|
|
|
||
|
|
all_passed = all(results)
|
||
|
|
if all_passed:
|
||
|
|
print("✓ 所有检查通过！打包成功。")
|
||
|
|
print("\n下一步:")
|
||
|
|
print(" 1. 将 dist/start_parse 文件夹复制到目标电脑")
|
||
|
|
print(" 2. 连接 EEG 设备并运行 start_parse.exe")
|
||
|
|
print(" 3. 观察控制台输出是否正常")
|
||
|
|
else:
|
||
|
|
print("✗ 部分检查未通过，请查看上述详细信息")
|
||
|
|
|
||
|
|
return all_passed
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|