Python中用keyboard库实现监听用户输入
前言
在用Python做一个Selenium网页自动化工具时,发现如果用户提供的数据太多,需要很长时间才能做完,有时希望中途停止。那么要增加一个判断,如果用户按下了某个组合键,则优雅地退出(要把当前正在处理的任务完成,在下一个任务开始之前退出)。
先是研究了回调函数(callback), 发现是能够实现的(那是当然,哈哈), 但是比较复杂。后来发现其实用keyboard库中的add_hotkey() 可以轻松实现这一功能
一、keyboard库
如果用pip install keyboard不成功,可以到这里下载whl文件。
Link
二、使用示例
1.引入库
代码如下(示例):
import keyboard
import time
def setStopFlag():
global blStop
blStop = True
if __name__ == '__main__':
blStop = False
# set hot key
keyboard.add_hotkey('ctrl+shift+t',setStopFlag)
print('----------Start-----------------')
intCnt = 0
while intCnt < 10 and blStop == False:
print( "%s Print result from loop in main process, intCnt is: %d"%(time.ctime(),intCnt))
time.sleep(5)
print("now wake up!")
intCnt +=1
if blStop == False:
print( 'Timeout!')
else:
print( 'End: stopped by user!')
设置一个退出标志, 在主功能中,判断标志是否被设置。设置了则退出。
调用add_hotkey()后,keyboard在后台监听. 当用户按下此组合键后,退出标志即可被设置。
整个功能是非阻塞的。