在 Python 中接管键盘中断信号的实现方法


Posted in Python onFebruary 04, 2020

在 Python 中接管键盘中断信号的实现方法

假设有这样一个需求,你需要从 Redis 中持续不断读取数据,并把这些数据写入到 MongoDB 中。你可能会这样写代码:

import json 
import redis 
import pymongo 
client = redis.Redis() 
handler = pymongo.MongoClient().example.col 
while True: 
  data_raw = client.blpop('data', timeout=300) 
  if not data_raw: 
    continue 
  data = json.loads(data_raw[1].decode()) 
  handler.insert_one(data)

但这样写有一个问题,就是每来一条数据都要连接一次 MongoDB,大量时间浪费在了网络 I/O上。

于是大家会把代码改成下面这样:

import json 
import redis 
import pymongo 
client = redis.Redis() 
handler = pymongo.MongoClient().example.col 
to_be_insert = [] 
while True: 
  data_raw = client.blpop('data', timeout=300) 
  if not data_raw: 
    continue 
  data = json.loads(data_raw[1].decode()) 
  to_be_insert.append(data) 
  if len(to_be_insert) >= 1000: 
    handler.insert_many(to_be_insert) 
    to_be_insert = []

每凑够1000条数据,批量写入到 MongoDB 中。

现在又面临另外一个问题。假设因为某种原因,我需要更新这个程序,于是我按下了键盘上的Ctrl + C强制关闭了这个程序。而此时to_be_insert列表里面有999条数据将会永久丢失——它们已经被从 Redis 中删除了,但又没有来得及写入 MongoDB 中。

我想实现,当我按下 Ctrl + C 时,程序不再从 Redis 中读取数据,但会先把to_be_insert中的数据(无论有几条)都插入 MongoDB 中。最后再关闭程序。

要实现这个需求,就必须在我们按下Ctrl + C时,程序还能继续运行一段代码。可问题是按下Ctrl + C时,程序就直接结束了,如何还能再运行一段代码?

实际上,当我们按下键盘上的Ctrl + C时,Python 收到一个名为SIGINT的信号。具体规则可以阅读官方文档。收到信号以后,Python 会调用一个信号回调函数。只不过默认的回调函数就是让程序抛出一个 KeyboardInterrupt异常导致程序关闭。现在,我们可以设法让 Python 使用我们自定义的一段函数来作为信号回调函数。

要使用信号,我们需用导入 Python 的signal库。然后自定义一个信号回调函数,当 Python 收到某个信号时,调用这个函数。

所以我们修改一下上面的代码:

import signal 
import json 
import redis 
import pymongo 
 
 
client = redis.Redis() 
handler = pymongo.MongoClient().example.col 
stop = False 
 
 
def keyboard_handler(signum, frame): 
  global stop 
  stop = True 
 
 
signal.signal(signal.SIGINT, keyboard_handler) 
 
to_be_insert = [] 
while not stop: 
  data_raw = client.blpop('data', timeout=300) 
  if not data_raw: 
    continue 
  data = json.loads(data_raw[1].decode()) 
  to_be_insert.append(data) 
  if len(to_be_insert) >= 1000: 
    handler.insert_many(to_be_insert) 
    to_be_insert = [] 
 
if to_be_insert: 
  handler.insert_many(to_be_insert)

我们定义了一个全局变量stop,默认为 False,所以默认情况下,while not stop所在的循环体会持续运行。

我们定义了一个函数keyboard_handler,它的作用是修改全局变量stop为 True。需要注意的是,在函数里面修改全局变量,必须先使用global 变量名声明这个变量为全局变量。否则无法修改。

修改以后,while not stop循环停止,于是程序进入:

if to_be_insert: 
  handler.insert_many(to_be_insert)

只要列表里面有数据,就会批量插入 MongoDB 中。然后程序结束。

整段代码的关键就在signal.signal(signal.SIGINT, keyboard_handler)这里把信号SIGINT与函数keyboard_handler关联上了,于是,在上面这段代码运行的任何时候,只要按下键盘的Ctrl + C,程序就会进入keyboard_handler函数里面,优先执行这个函数里面的代码。执行完成以后,回到之前中断的地方,继续执行之前没有完成的代码。而由于在函数里面我已经修改了stop的值,所以原来的循环不能继续执行,于是进入最后的收尾工作。

需要注意的是,如果你的整个代码全都是使用 Python 写的,那么 signal可以在你程序的任何阶段触发,只要你按下 Ctrl + C,立刻就会进入设置好的信号回调函数中。

但如果你的代码中,有一部分代码是使用 C 语言写的,那么当你按下Ctrl + C以后,可能需要等这段C 语言的代码运行完成以后,才会进入你设置的信号回调函数中。

总结

以上所述是小编给大家介绍的在 Python 中接管键盘中断信号的处理方法,希望对大家有所帮助!

Python 相关文章推荐
python urllib爬取百度云连接的实例代码
Jun 19 Python
从CentOS安装完成到生成词云python的实例
Dec 01 Python
微信跳一跳python自动代码解读1.0
Jan 12 Python
python中的随机函数random的用法示例
Jan 27 Python
python2和python3的输入和输出区别介绍
Nov 20 Python
Python编程深度学习计算库之numpy
Dec 28 Python
Python多线程threading模块用法实例分析
May 22 Python
python实现的生成word文档功能示例
Aug 23 Python
pygame库实现俄罗斯方块小游戏
Oct 29 Python
Python格式化输出--%s,%d,%f的代码解析
Apr 29 Python
python线程里哪种模块比较适合
Aug 02 Python
一小时学会TensorFlow2之基本操作2实例代码
Sep 04 Python
在TensorFlow中屏蔽warning的方式
Feb 04 #Python
Python和Anaconda和Pycharm安装教程图文详解
Feb 04 #Python
Python3.7黑帽编程之病毒篇(基础篇)
Feb 04 #Python
python with (as)语句实例详解
Feb 04 #Python
Python实现实时数据采集新型冠状病毒数据实例
Feb 04 #Python
在tensorflow中实现屏蔽输出的log信息
Feb 04 #Python
Python变量作用域LEGB用法解析
Feb 04 #Python
You might like
PHP之浮点数计算比较以及取整数不准确的解决办法
2015/07/29 PHP
yii2.0数据库迁移教程【多个数据库同时同步数据】
2016/10/08 PHP
PHP实现无限分类的实现方法
2016/11/14 PHP
一些mootools的学习资源
2010/02/07 Javascript
JavaScript中关于indexOf的使用方法与问题小结
2010/08/05 Javascript
两种方法实现文本框输入内容提示消失
2013/03/17 Javascript
jquery(hide方法)隐藏指定元素实例
2013/11/11 Javascript
JS Loading功能的简单实现
2013/11/29 Javascript
jQuery操作元素css样式的三种方法
2014/06/04 Javascript
jQuery实现表格展开与折叠的方法
2015/05/04 Javascript
浅谈下拉菜单中的Option对象
2015/05/10 Javascript
基于JavaScript实现生成名片、链接等二维码
2015/09/20 Javascript
JavaScript使ifram跨域相互访问及与PHP通信的实例
2016/03/03 Javascript
js实现股票实时刷新数据案例
2017/05/14 Javascript
微信通过页面(H5)直接打开本地app的解决方法
2017/09/09 Javascript
javascript实现导航栏分页效果
2019/06/27 Javascript
[01:29]2017 DOTA2国际邀请赛官方英雄手办展示
2017/03/18 DOTA
Python中运行并行任务技巧
2015/02/26 Python
Python元组拆包和具名元组解析实例详解
2018/03/26 Python
python实现自动发送报警监控邮件
2018/06/21 Python
在Python中使用turtle绘制多个同心圆示例
2019/11/23 Python
css3学习之2D转换功能详解
2016/12/23 HTML / CSS
HTML5 video 事件应用示例
2014/09/11 HTML / CSS
HTML5新增的8类INPUT输入类型介绍
2015/07/06 HTML / CSS
英国知名的皮手套品牌:Dents
2016/11/13 全球购物
当一个对象被当作参数传递到一个方法后,此方法可改变这个对象的属性,并可返回变化后的结果,那么这里到底是值传递还是引用传递?
2014/09/09 面试题
90后毕业生的求职信范文
2013/09/21 职场文书
网络工程专业毕业生推荐信
2013/10/28 职场文书
给朋友的道歉信
2014/01/09 职场文书
《在家里》教后反思
2014/03/01 职场文书
大专毕业生求职信
2014/07/05 职场文书
村安全生产责任书
2014/08/25 职场文书
企业计划生育责任书
2015/05/09 职场文书
2016银行招聘自荐信
2016/01/28 职场文书
Mysql 如何查询时间段交集
2021/06/08 MySQL
Python+OpenCV实现在图像上绘制矩形
2022/03/21 Python