惊雷无声 发表于 2024-5-18 04:11:42

用Python步伐如何捕捉Ctrl+C终止信号?

对于一些连续运行或者长时间运行的Python步伐而言,如服务器的后端,或者是长时间运行的科学盘算步伐。
当我们涉及到一些中途退出的操作时,比如利用Ctrl+C来退出正在运行的步伐。
这种场景的出现一般有两个可能性:一是步伐出现了问题,必要终止步伐来对其进行调整。另一种是步伐自己是正确的,但是步伐运行的速度太慢了,也有可能是想提前结束,这种场景下很多时间我们是希望可以保留其相应的盘算结果的。
但是假如我们利用的是一些第三方的数据存储格式来存储数据,不一定可以支持连续的存储,非经常见的是在步伐执行结束之后,再将结果进行生存。但是由于步伐被提前终止了,此时就必要一些特殊的手段来对中途终止的步伐的结果进行生存。
基础案例

我们先来看一个比较简单的案例:一个普通的打印数字的步伐,每隔1s的时间就打印一个数字出来,我们可以利用python的signal.signal来捕捉这个终止信号。
# signal_exit.py
import signal
import sys

def signal_handler(signal, frame):
    print ('\nSignal Catched! You have just type Ctrl+C!')
    sys.exit(0)

if __name__ == '__main__':
    import time

    signal.signal(signal.SIGINT, signal_handler)
    for x in range(100):
      time.sleep(1)

      print (x)当我们运行这个步伐到一半时,同时按下Ctrl+C,我们会得到如下的结果:
$ python3 signal_exit.py
0
1
2
^C

Signal Catched! You have just type Ctrl+C!这个结果表明,我们在步伐运行的过程中捕捉到了Ctrl+C的这个外部操作,并且对该操作进行了相应的处置惩罚之后,才终止了步伐的运行。必要注意的是,假如此时不加上sys.exit(0)这个终止的操作,这个步伐不会被制止,会继承运行下去,相当于只是捕捉了异常终止信号但不做任何的处置惩罚。
给终止信号传入外部参数

在上面的一个案例中,仅仅只是捕捉了“终止运行”的这个外部信号,但是假如更进一步的,我们想捕捉到最后一个输出的数字是多少,这个时间要如何操作呢?signal.signal函数自己并不支持很多的参数传入,此时建议采取的是自行创建一个类,将signal_handler函数封装为类的成员函数,这样我们就可以获取到相应的内部参数,如下面这个案例所示:
# signal_exit.py
import signal
import sys
import time

class Printer:
    def __init__(self):
      self.x = 0
      signal.signal(signal.SIGINT, self.signal_handler)

    def signal_handler(self, signal, frame):
      print ('\nSignal Catched! You have just type Ctrl+C! The last number is: {}'.format(self.x))
      sys.exit(0)
        #学习中遇到问题没人解答?小编创建了一个Python学习交流群:153708845

    def run(self, counter=10):
      while self.x < counter:
            print (self.x)
            time.sleep(1)
            self.x += 1

if __name__ == '__main__':
    printer = Printer()

    printer.run(counter=100)此时假如在步伐正在运行的状态下同时按下Ctrl+C,得到的结果如下:
$ python3 signal_exit.py
0
1
2
3
^C

Signal Catched! You have just type Ctrl+C! The last number is: 3可以看到,我们成功的捕捉到了最后一个被输出出来的参数。
总结

当我们准备去杀死一个进程时,从步伐设计的自己来思量,我们应当要设计一定的保护方案来确保步伐被非正常终止时,相应的盘算结果也能够被很好的生存下来。在Python中可以利用signal.signal函数来实现这样的功能,但是假如要实现数据的生存功能,必要团结一个实际的类来实现。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: 用Python步伐如何捕捉Ctrl+C终止信号?