在SIGINT后保存工作

Question

我有一个程序，运行起来很耗时间。我希望它能在我按下 ctrl-c 的时候，捕捉到这个信号，然后调用 self.save_work() 方法来保存工作进度。

目前我的 signal_handler() 方法不能正常工作，因为在程序执行到 signal_handler() 的时候，self 还没有被定义。

我该怎么做才能确保在收到 SIGINT 信号后，能够调用 self.save_work 呢？

#!/usr/bin/env python
import signal 

def signal_handler(signal, frame):    
    self.save_work()   # Does not work
    exit(1)
signal.signal(signal.SIGINT, signal_handler)

class Main(object):
    def do_stuff(self):
        ...
    def save_work(self):
        ...
    def __init__(self):
        self.do_stuff()
        self.save_work()

if __name__=='__main__':
    Main()

Answer 1

在编程中，有时候我们需要把一些代码放在特定的地方，以便在需要的时候调用。这就像在厨房里把调料放在容易拿到的地方，做菜的时候就能快速找到。

有些人可能会遇到一个问题，就是在调用这些代码时，发现它们并没有按预期工作。这可能是因为代码的位置不对，或者调用的方式不对，就像你在找调料时，发现它们被放在了不容易找到的地方。

解决这个问题的方法通常是检查代码的放置位置，确保它们在正确的地方，并且调用的方式也是正确的。这样一来，当你需要使用这些代码时，就能顺利地找到并使用它们。

import signal

def signal_handler(signal, frame):    
   #do some stuff

def main():
   #do some more stuff


if __name__=='__main__':
    signal.signal(signal.SIGINT, signal_handler)
    main()

Answer 2

通常来说，“工作”会涉及到某种大循环。为了控制你的循环，防止它在某个未知的步骤中出错，你可以使用下面这个上下文管理器：

import signal

class GracefulInterruptHandler(object):

    def __init__(self, sig=signal.SIGINT):
        self.sig = sig

    def __enter__(self):

        self.interrupted = False
        self.released = False

        self.original_handler = signal.getsignal(self.sig)

        def handler(signum, frame):
            self.release()
            self.interrupted = True

        signal.signal(self.sig, handler)

        return self

    def __exit__(self, type, value, tb):
        self.release()

    def release(self):

        if self.released:
            return False

        signal.signal(self.sig, self.original_handler)

        self.released = True

        return True

使用方法：

import time

/// do stuff:
with GracefulInterruptHandler() as h:
    for i in xrange(1000):
        print "..."
        time.sleep(1)
        if h.interrupted:
            print "interrupted!"
            time.sleep(5)
            break

save_work()

来源链接： https://gist.github.com/2907502

Answer 3

如果你只是想捕捉到按下 ctrl+c 的情况，那么你可以捕捉到 KeyboardInterrupt 这个异常：

class Main(object):
    def do_stuff(self):
        ...
    def save_work(self):
        ...
    def __init__(self):
        try:
            self.do_stuff()
        except KeyboardInterrupt:
            pass # Or print helpful info
        self.save_work()

不过我觉得这样设计并不好。看起来你应该使用一个函数，而不是构造函数。