如何通过键盘事件动态重置辐射剂量计算参数

本文介绍在python命令行程序中,使用简单输入指令(如“reset”或“quit”)中断循环、返回参数输入阶段的方法,无需复杂线程或gui,兼容现有计算逻辑。

在辐射剂量计算这类交互式批处理场景中,用户常需反复调整DTU型号、分组系数(K_group)或本底剂量(Dfon)等参数,而当前代码的 while True: 循环会持续等待 I1/I2 输入,缺乏退出或重置机制。直接引入 keyboard 库监听全局按键(如 Esc)虽可行,但存在跨平台兼容性差、需管理员权限、与 input() 阻塞冲突等问题。更稳健、简洁且符合命令行习惯的方案是:将特定字符串(如 "reset"、"quit")作为软中断指令,嵌入现有 input() 流程中

✅ 推荐实现:分层中断 + 指令驱动流程控制

修改核心逻辑,将原单层无限循环拆分为两级结构:

  • 外层循环:管理 DTU 选择与 K_group/K_dreif/Dfon 等全局参数;
  • 内层循环:处理实时 I1/I2 输入与剂量计算。

当用户在任一 input() 中输入预设指令时,立即跳出对应层级,实现精准跳转:

# ...(加载JSON、初始化colorama等前置代码保持不变)

# 主循环:管理DTU及基础参数
while True:
    print("\n" + "="*40)
    dtu_input = input("Номер ДТУ = ").strip()
    if dtu_input.lower() in ["quit", "exit"]:
        print("Выход из программы.")
        break

    try:
        dtu = int(dtu_input)
        # ...(原有DTU参数映射逻辑,保持不变)
    except ValueError:
        print(f"{Fore.RED}Ошибка: введите число (1, 2 или 3){Style.RESET_ALL}")
        continue

    # KdrDf(start) — 现在置于外层循环内
    while True:
        n_input = input("K_group = ").strip()
        if n_input.lower() == "reset":
            break  # 跳出内层,回到DTU选择
        if n_input.lower() in ["quit", "exit"]:
            exit(0)

        try:
            n = int(n_input)
            if not (0 <= n < len(k)):
                raise ValueError("Индекс вне диапазона")
            ks = k[n]
        except ValueError as e:
            print(f"{Fore.RED}Ошибка K_group: {e}{Style.RESET_ALL}")
            continue

        try:
            k_dr = float(input('K_dreif = ').strip().replace(',', '.'))
            dfon = float(input("Dfon = ").strip().replace(',', '.'))
        except ValueError:
            print(f"{Fore.RED}Ошибка ввода числа{Style.RESET_ALL}")
            continue

        # 计算循环(内层最内层)
        print(f"\nРасчет для DTU-{dtu}, K_group={n}. Введите 'reset' для смены параметров.")
        while True:
            i1_str = input("I1 = ").strip()
            if i1_str.lower() == "reset":
                break  # 跳出计算循环,回到K_group输入
            if i1_str.lower() in ["quit", "exit"]:
                exit(0)

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

            # 数值解析(保留原逗号→小数点转换)
            try:
                i_1 = float(i1_str.replace(',', '.'))
                i_2 = float(i2_str.replace(',', '.'))
                i1 = i_1 * ks
                i2 = i_2 * ks
                d1 = (k_dr * i1 * a) + b
                d2 = (k_dr * i2 * a) + b
                D = ((d1 + d2) / 2) - dfon

                # 根据D值更新a/b并输出(逻辑同原代码)
                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}")
            except ValueError:
                print(f"{Fore.RED}Некорректное значение I1 или I2{Style.RESET_ALL}")
                continue

⚠️ 关键注意事项

  • 指令大小写不敏感:使用 .lower() 统一处理,提升用户体验;
  • 避免阻塞冲突:input() 是同步阻塞调用,keyboard 库在此场景下反而增加复杂度和风险,应优先采用输入指令模式;
  • 错误恢复友好:每次 input() 后校验输入合法性,失败时 continue 而非崩溃,保证流程可控;
  • 清晰的视觉反馈:在进入新计算阶段时打印提示(如 "Расчет для DTU-1..."),明确当前上下文;
  • 安全退出:"quit"/"exit" 在任意层级均可终止程序,避免陷入死循环。

此方案零依赖新增库,完全兼容原逻辑,仅通过结构化 break 和 continue 实现灵活的参数重载,是命令行科学计算工具中轻量、可靠、易维护的交互增强实践。