如何“停止”和“恢复”长时间运行的Python脚本？

Question

我写了Python脚本来处理大量的大文本文件，并且可能 要花费很多时间
。有时，需要停止正在运行的脚本并在以后恢复它。停止脚本的可能原因是程序崩溃，磁盘空间不足或需要执行此操作的其他原因。我想为脚本实现一种“停止/恢复”机制。

• 在 车站 ：脚本退出和保存其当前状态。
• 在 简历 ：剧本开始，但是从最新的保存状态持续

我将使用 pickle 和 信号 模块来实现它。

我将很高兴听到如何以pythonic方式进行操作。

谢谢！

Answer 1

这很简单，希望可以对您有所帮助：

import time
import pickle


REGISTRY = None


def main(start=0):
    """Do some heavy work ..."""

    global REGISTRY

    a = start
    while 1:
        time.sleep(1)
        a += 1
        print a
        REGISTRY = pickle.dumps(a)


if __name__ == '__main__':
    print "To stop the script execution type CTRL-C"
    while 1:
       start = pickle.loads(REGISTRY) if REGISTRY else 0
        try:
            main(start=start)
        except KeyboardInterrupt:
            resume = raw_input('If you want to continue type the letter c:')
            if resume != 'c':
                break

运行示例：

$ python test.py
To stop the script execution type CTRL-C
1
2
3
^CIf you want to continue type the letter c:c
4
5
6
7
8
9
^CIf you want to continue type the letter c:
$ python test.py