如何通过控制台按键在独立线程中中断函数的自运行？——以按‘q’终止含time.sleep的递归main函数为例

这问题我熟，刚好之前处理过类似的场景！要实现按'q'键中断递归运行的函数，核心思路是用一个线程安全的终止标志，配合独立线程监听键盘输入，具体实现步骤如下：

1. 使用线程安全的终止信号 #

推荐用threading.Event作为终止标志，它是Python内置的线程安全同步原语，比普通全局变量更可靠，能避免多线程下的竞态问题。

2. 编写按键监听线程 #

创建一个独立线程，专门监听控制台输入，当检测到用户输入q时，触发终止信号。

3. 修改递归函数以响应终止信号 #

在递归函数的每次循环/递归前，检查终止信号是否被触发，如果触发就停止递归，退出函数。

完整代码示例 #

import threading
import time

# 创建线程安全的终止事件
stop_event = threading.Event()

def key_listener():
    """监听控制台输入，按下q时触发终止信号"""
    while not stop_event.is_set():
        # 读取输入（如果需要无阻塞监听，可改用getch库，后面有说明）
        user_input = input().strip().lower()
        if user_input == 'q':
            print("收到终止信号，准备停止运行...")
            stop_event.set()

def main():
    """递归运行的目标函数"""
    # 检查终止信号，如果已触发则停止递归
    if stop_event.is_set():
        return
    
    print("函数正在运行...（按q键终止）")
    # 用Event.wait替代time.sleep，可被终止信号及时唤醒，不用等满10秒
    if not stop_event.wait(10):
        # 等待10秒后无终止信号，继续递归
        main()

if __name__ == "__main__":
    # 启动按键监听线程，设置为守护线程，主程序退出时自动结束
    listener_thread = threading.Thread(target=key_listener, daemon=True)
    listener_thread.start()
    
    # 启动递归函数
    main()
    
    print("函数已停止运行")

额外说明 #

• 如果你的场景需要无阻塞监听按键（比如不想被input()阻塞程序后台运行），可以用getch库（先执行pip install getch安装），修改监听函数如下：

  import getch
  def key_listener():
      while not stop_event.is_set():
          key = getch.getch().lower()
          if key == 'q':
              print("收到终止信号，准备停止运行...")
              stop_event.set()
  

• 用stop_event.wait(10)替代time.sleep(10)的好处是，当终止信号触发时，程序会立刻从等待中醒来，不需要等满10秒才停止，响应更及时。

内容的提问来源于stack exchange，提问作者Vyacheslav