如何通过键盘事件中断计算循环并重置参数输入

本文介绍在python命令行程序中,使用简单字符串输入(如"reset"或"quit")模拟键盘事件,实现在计算循环中动态中断、返回初始参数设置环节,提升交互灵活性与用户体验。

在您的辐射剂量计算程序中,核心需求是:在持续运行的 while True 计算循环中,不依赖外部库监听全局按键(如 keyboard.is_pressed()),也能快速退出当前流程、重新选择 DTU 编号并重载参数。由于 input() 是阻塞式调用,直接使用 keyboard 库监听(如 keyboard.add_hotkey('esc', ...))在标准终端中常因焦点/权限问题失效,且会引入线程复杂性。因此,推荐采用轻量、可靠、零依赖的“语义化指令输入”方案——即约定特定字符串(如 "reset"、"quit"、"back")作为软中断信号。

✅ 推荐实现方式(结构清晰、兼容性强)

将原有单层 while True 拆分为外层主循环(DTU 选择) + 内层计算循环(I1/I2 输入),并在关键 input() 处添加指令判断:

# ... [前面的 JSON 加载、系数初始化等保持不变] ...

while True:  # ← 外层循环:控制 DTU 选择与参数重载
    print("\n" + "="*40)
    dtu_input = input("Номер ДТУ (1-3) 或输入 'quit' 退出程序: ").strip()

    if dtu_input.lower() == "quit":
        print("Программа завершена.")
        break

    try:
        dtu = int(dtu_input)
        if dtu not in (1, 2, 3):
            print(f"{Fore.RED}Ошибка: введите 1, 2 или 3.{Style.RESET_ALL}")
            continue
    except ValueError:
        print(f"{Fore.RED}Ошибка: введите число или 'quit'.{Style.RESET_ALL}")
        continue

    # 根据 DTU 加载对应参数(原逻辑)
    if dtu == 1:
        a_m = [a1_values[0], a2_values[0], a3_values[0]]
        b_m = [b1_values[0], b2_values[0], b3_values[0]]
        k = k_sens1
    elif dtu == 2:
        a_m = [a1_values[1], a2_values[1], a3_values[1]]
        b_m = [b1_values[1], b2_values[1], b3_values[1]]
        k = k_sens2
    else:  # dtu == 3
        a_m = [a1_values[2], a2_values[2], a3_values[2]]
        b_m = [b1_values[2], b2_values[2], b3_values[2]]
        k = k_sens3

    # KdrDf 参数输入(支持 reset)
    while True:
        n_input = input("K_group (0-6) 或输入 'reset' 返回选择 DTU: ").strip()
        if n_input.lower() == "reset":
            break  # ← 退出内层循环,自动回到外层 DTU 选择
        try:
            n = int(n_input)
            if not 0 <= n <= 6:
                print(f"{Fore.YELLOW}Предупреждение: K_group вне диапазона 0–6. Используется по умолчанию 0.{Style.RESET_ALL}")
                n = 0
            ks = k[n]
        except ValueError:
            print(f"{Fore.RED}Ошибка: введите число или 'reset'.{Style.RESET_ALL}")
            continue
        break  # 成功输入 K_group,跳出参数输入循环

    k_dr = float(input('K_dreif = '))
    dfon = float(input("Dfon = "))

    # 主计算循环(支持实时 reset)
    while True:
        i1_str = input("I1 = (введите 'reset' для сброса параметров, 'quit' для выхода): ").strip()
        if i1_str.lower() == "quit":
            exit(0)
        if i1_str.lower() == "reset":
            break  # ← 中断计算,跳回 K_group 输入(或直接到 DTU 选择,取决于设计)

        i2_str = input("I2 = ").strip()
        if i2_str.lower() == "quit":
            exit(0)
        if i2_str.lower() == "reset":
            break

        # 处理小数点替换与计算(原逻辑)
        i1_str = i1_str.replace(',', '.')
        i2_str = i2_str.replace(',', '.')
        try:
            i_1, i_2 = float(i1_str), float(i2_str)
        except ValueError:
            print(f"{Fore.RED}Ошибка: введите корректные числа.{Style.RESET_ALL}")
            continue

        i1, i2 = i_1 * ks, i_2 * ks
        d1 = (k_dr * i1 * a_m[0]) + b_m[0]
        d2 = (k_dr * i2 * a_m[0]) + b_m[0]
        D = ((d1 + d2) / 2) - dfon

        # 自适应参数更新(原逻辑)
        if 1.5 <= D < 15:
            a, b = a_m[1], b_m[1]
            d1 = (k_dr * i1 * a) + b
            d2 = (k_dr * i2 * a) + b
            D = ((d1 + d2) / 2) - dfon
            print(f"{Fore.YELLOW}{round(D, 3)}{Style.RESET_ALL}")
        elif D >= 15:
            a, b = a_m[2], b_m[2]
            d1 = (k_dr * i1 * a) + b
            d2 = (k_dr * i2 * a) + b
            D = ((d1 + d2) / 2) - dfon
            print(f"{Fore.RED}{round(D, 3)}{Style.RESET_ALL}")
        else:
            print(f"{Fore.GREEN}{round(D, 3)}{Style.RESET_ALL}")

⚠️ 注意事项与最佳实践

  • 避免 keyboard 库的常见陷阱:keyboard 需管理员权限(Windows)、易与 IDE 终端冲突,且 input() 阻塞时无法响应;本方案纯 Python 标准库,100% 兼容。
  • 指令统一性:建议固定使用 reset(返回上一级)、quit(完全退出),全部转为 .lower() 处理,提升鲁棒性。
  • 错误防御增强:对 int()/float() 转换增加 try-except,防止崩溃;对越界索引提供默认值或提示。
  • 用户体验优化:添加分隔线 ===、明确提示(如 "или 'reset'"),让用户清楚当前可执行的操作。
  • 扩展性提示:若未来需真正热键(如 Ctrl+C 中断当前计算但不清空 DTU),可用 signal.signal(signal.SIGINT, ...) 捕获 KeyboardInterrupt,但需重构为状态机模式。

此方案无需额外安装、无兼容性风险,用最少改动达成您所需的“键盘事件式交互”,让辐射剂量计算流程真正具备现场调试的灵活性。