如何防止代码块被Python中的KeyboardInterrupt中断？

Question

我正在编写一个程序，通过pickle模块缓存一些结果。目前发生的情况是，如果在执行dump操作时按ctrl-c，则会dump被打断，从而导致文件损坏（即仅部分写入，因此无法load再次进行编辑）。

有没有办法使dump代码块或通常的代码块不间断？我当前的解决方法如下所示：

try:
  file = open(path, 'w')
  dump(obj, file)
  file.close()
except KeyboardInterrupt:
  file.close()
  file.open(path,'w')
  dump(obj, file)
  file.close()
  raise

如果操作被中断，重新启动该操作似乎很愚蠢，所以我正在寻找一种推迟中断的方法。我该怎么做呢？

Answer 1

将函数放在线程中，然后等待线程完成。

除非使用特殊的C api，否则无法中断Python线程。

import time
from threading import Thread

def noInterrupt():
    for i in xrange(4):
        print i
        time.sleep(1)

a = Thread(target=noInterrupt)
a.start()
a.join()
print "done"


0
1
2
3
Traceback (most recent call last):
  File "C:\Users\Admin\Desktop\test.py", line 11, in <module>
    a.join()
  File "C:\Python26\lib\threading.py", line 634, in join
    self.__block.wait()
  File "C:\Python26\lib\threading.py", line 237, in wait
    waiter.acquire()
KeyboardInterrupt

看看如何将中断推迟到线程完成为止？

在这里它适合您的使用：

import time
from threading import Thread

def noInterrupt(path, obj):
    try:
        file = open(path, 'w')
        dump(obj, file)
    finally:
        file.close()

a = Thread(target=noInterrupt, args=(path,obj))
a.start()
a.join()