采集卡bug修复
This commit is contained in:
145
main_multi.py
Normal file
145
main_multi.py
Normal file
@@ -0,0 +1,145 @@
|
||||
"""
|
||||
多配置组启动器
|
||||
支持同时启动多个配置组的自动化程序
|
||||
"""
|
||||
import multiprocessing
|
||||
import sys
|
||||
from config import config_manager
|
||||
from main_single import run_automation_for_group
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
# 重新加载配置
|
||||
config_manager.load_config()
|
||||
|
||||
# 获取所有配置组
|
||||
groups = config_manager.config.get('groups', [])
|
||||
|
||||
if not groups:
|
||||
print("❌ 没有找到任何配置组")
|
||||
print("请先运行 gui_config.py 创建配置组")
|
||||
return
|
||||
|
||||
# 询问要启动哪些配置组
|
||||
print("=" * 60)
|
||||
print("🔥 多配置组启动器")
|
||||
print("=" * 60)
|
||||
print("\n可用配置组:")
|
||||
for i, group in enumerate(groups):
|
||||
active_mark = "✓" if group.get('active', False) else " "
|
||||
print(f" [{i}] {active_mark} {group['name']}")
|
||||
print(f" 串口: {group['serial_port']} | 采集卡: {group['camera_index']}")
|
||||
|
||||
print("\n选择启动方式:")
|
||||
print(" 1. 启动所有活动配置组")
|
||||
print(" 2. 启动所有配置组")
|
||||
print(" 3. 选择特定配置组")
|
||||
print(" 0. 退出")
|
||||
|
||||
choice = input("\n请选择 (0-3): ").strip()
|
||||
|
||||
selected_indices = []
|
||||
|
||||
if choice == "0":
|
||||
print("👋 退出")
|
||||
return
|
||||
elif choice == "1":
|
||||
# 启动所有活动配置组
|
||||
selected_indices = [i for i, g in enumerate(groups) if g.get('active', False)]
|
||||
if not selected_indices:
|
||||
print("❌ 没有活动的配置组")
|
||||
return
|
||||
print(f"\n✅ 将启动 {len(selected_indices)} 个活动配置组")
|
||||
elif choice == "2":
|
||||
# 启动所有配置组
|
||||
selected_indices = list(range(len(groups)))
|
||||
print(f"\n✅ 将启动所有 {len(selected_indices)} 个配置组")
|
||||
elif choice == "3":
|
||||
# 选择特定配置组
|
||||
indices_input = input("请输入要启动的配置组索引(用逗号分隔,如: 0,1,2): ").strip()
|
||||
try:
|
||||
selected_indices = [int(x.strip()) for x in indices_input.split(',')]
|
||||
# 验证索引有效性
|
||||
selected_indices = [i for i in selected_indices if 0 <= i < len(groups)]
|
||||
if not selected_indices:
|
||||
print("❌ 没有有效的配置组索引")
|
||||
return
|
||||
print(f"\n✅ 将启动 {len(selected_indices)} 个配置组")
|
||||
except ValueError:
|
||||
print("❌ 输入格式错误")
|
||||
return
|
||||
else:
|
||||
print("❌ 无效选择")
|
||||
return
|
||||
|
||||
# 显示将要启动的配置组
|
||||
print("\n将要启动的配置组:")
|
||||
for idx in selected_indices:
|
||||
group = groups[idx]
|
||||
print(f" • {group['name']} (串口:{group['serial_port']}, 采集卡:{group['camera_index']})")
|
||||
|
||||
confirm = input("\n确认启动? (y/n): ").strip().lower()
|
||||
if confirm != 'y':
|
||||
print("❌ 取消启动")
|
||||
return
|
||||
|
||||
# 启动多进程
|
||||
print("\n🚀 开始启动多个配置组...")
|
||||
processes = []
|
||||
|
||||
for idx in selected_indices:
|
||||
group = groups[idx]
|
||||
print(f"启动进程: {group['name']}...")
|
||||
process = multiprocessing.Process(
|
||||
target=run_automation_for_group,
|
||||
args=(idx,),
|
||||
name=f"Group-{idx}-{group['name']}"
|
||||
)
|
||||
process.start()
|
||||
processes.append((idx, group['name'], process))
|
||||
print(f"✅ {group['name']} 已启动 (PID: {process.pid})")
|
||||
|
||||
print(f"\n✅ 成功启动 {len(processes)} 个配置组进程")
|
||||
print("\n" + "=" * 60)
|
||||
print("运行状态:")
|
||||
print("=" * 60)
|
||||
|
||||
# 监控进程状态
|
||||
try:
|
||||
while True:
|
||||
alive_count = 0
|
||||
for idx, name, proc in processes:
|
||||
if proc.is_alive():
|
||||
alive_count += 1
|
||||
else:
|
||||
print(f"⚠️ {name} 进程已退出 (退出码: {proc.exitcode})")
|
||||
|
||||
if alive_count == 0:
|
||||
print("\n所有进程已退出")
|
||||
break
|
||||
|
||||
import time
|
||||
time.sleep(2)
|
||||
|
||||
# 打印存活状态
|
||||
alive_names = [name for idx, name, proc in processes if proc.is_alive()]
|
||||
if alive_names:
|
||||
print(f"\r📊 运行中: {', '.join(alive_names)} ({alive_count}/{len(processes)})", end='', flush=True)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n🛑 收到停止信号,正在关闭所有进程...")
|
||||
for idx, name, proc in processes:
|
||||
if proc.is_alive():
|
||||
print(f"正在停止 {name}...")
|
||||
proc.terminate()
|
||||
proc.join(timeout=5)
|
||||
if proc.is_alive():
|
||||
print(f"强制停止 {name}...")
|
||||
proc.kill()
|
||||
print(f"✅ {name} 已停止")
|
||||
print("\n👋 所有进程已停止")
|
||||
|
||||
if __name__ == "__main__":
|
||||
multiprocessing.freeze_support() # Windows下需要
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user