Python中如何通过按键在循环步骤中退出或修改变量

原代码的核心问题 #

1. 按键判断逻辑错误：pynput捕获的字符按键是KeyCode对象，直接和字符串'd'比较永远不成立，需要用keyboard.KeyCode.from_char('d')来匹配
2. 全局变量线程不安全：直接修改全局布尔变量可能出现线程竞争，改用threading.Event更可靠
3. 阻塞sleep无法响应按键：time.sleep(10)会阻塞线程，期间无法检测按键状态，需要拆分成长度更短的循环sleep，每次检查标志
4. 线程内退出方式错误：在listener线程调用quit()会导致异常，应该用事件通知主线程退出

改进后的实现代码 #

import os
import time
import threading
from pynput import keyboard

# 用线程安全的Event代替全局布尔变量
delete_event = threading.Event()
kill_event = threading.Event()

def on_key_press(key):
    try:
        # 匹配'd'键，切换删除状态
        if key == keyboard.KeyCode.from_char('d'):
            if delete_event.is_set():
                delete_event.clear()
            else:
                delete_event.set()
    except AttributeError:
        # 处理特殊按键
        if key == keyboard.Key.esc:
            kill_event.set()
            return False  # 停止listener

def process_file(file):
    # 模拟步骤1：展示文件+处理
    show_file(file)
    file = foo1(file)
    # 拆分sleep，每0.1秒检查一次事件
    for _ in range(100):
        if kill_event.is_set():
            return False
        if delete_event.is_set():
            return True
        time.sleep(0.1)
    
    # 步骤2
    file = foo2(file)
    for _ in range(100):
        if kill_event.is_set():
            return False
        if delete_event.is_set():
            return True
        time.sleep(0.1)
    
    # 步骤3
    file = foo3(file)
    for _ in range(100):
        if kill_event.is_set():
            return False
        if delete_event.is_set():
            return True
        time.sleep(0.1)
    
    # 最终判断：如果没有触发删除，生成摘要
    return False

def loop():
    file_list = ["test1.html", "test2.html"]  # 替换为你的文件列表
    for file in file_list:
        if kill_event.is_set():
            break
        
        # 重置删除事件，处理当前文件前清空状态
        delete_event.clear()
        
        need_delete = process_file(file)
        if kill_event.is_set():
            break
        
        if need_delete:
            delete(file)
            print(f"已删除文件: {file}")
        else:
            # 生成并写入摘要
            summary = generate_summary(file)  # 替换为你的摘要生成逻辑
            write_file(summary)
            print(f"已生成文件摘要: {file}")
        close_file(file)

# 以下为模拟的辅助函数，替换为你的实际逻辑
def show_file(file):
    print(f"正在展示文件: {file}")

def foo1(file):
    print(f"执行步骤1处理: {file}")
    return file

def foo2(file):
    print(f"执行步骤2处理: {file}")
    return file

def foo3(file):
    print(f"执行步骤3处理: {file}")
    return file

def delete(file):
    print(f"模拟删除文件: {file}")

def generate_summary(file):
    return f"{file} 的摘要内容"

def write_file(summary):
    print(f"写入摘要: {summary}")

def close_file(file):
    print(f"关闭文件: {file}")

if __name__ == "__main__":
    # 启动按键监听线程
    listener = keyboard.Listener(on_press=on_key_press)
    listener.start()
    
    # 启动文件处理线程
    process_thread = threading.Thread(target=loop)
    process_thread.start()
    
    # 等待处理线程结束
    process_thread.join()
    # 停止监听
    listener.stop()

关键改进说明 #

• 线程安全事件：threading.Event的set()、clear()、is_set()方法都是线程安全的，避免多线程下的变量竞争问题
• 非阻塞sleep：把10秒的长sleep拆成100次0.1秒的短sleep，每次sleep后立即检查事件状态，实现近乎实时的按键响应
• 按键匹配修正：正确处理字符按键和特殊按键，避免无效的按键判断
• 清晰的流程控制：每个处理步骤都优先检查退出事件，确保按下esc能快速终止程序；处理完每个文件后重置删除事件，避免状态污染下一个文件

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