Jak zapobiec blokowaniu kodu przez KeyboardInterrupt w Pythonie?

Question

Piszę program, który buforuje niektóre wyniki przez moduł pikle. Co się dzieje w tej chwili jest to, że jeśli uderzę CTRL-C podczas operacji dump występuje, dump zostaje przerwany i wynikowy plik jest uszkodzony (czyli tylko częściowo napisany, więc to nie może być load ed ponownie.

Czy istnieje sposób, aby dump, albo w ogóle bloku kodu, żródło Moje bieżące obejście wygląda mniej więcej tak:?

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

wydaje się głupie, aby ponownie uruchomić operację, jeśli zostanie przerwane, więc szukam sposobu przerwać przerwanie Jak to zrobić?

Answer 1

Umieść funkcję w wątku i poczekaj na zakończenie wątku.

Nici w języku Python nie mogą być przerywane, z wyjątkiem specjalnej aplikacji C api.

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 

Zobacz, jak przerwanie zostało odroczone do czasu zakończenia wątku?

Tutaj jest przystosowany do użytku:

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() 

Answer 2

użyć modułu signal wyłączyć SIGINT na czas trwania procesu:

s = signal.signal(signal.SIGINT, signal.SIG_IGN) 
do_important_stuff() 
signal.signal(signal.SIGINT, s) 

Answer 3

Moim zdaniem przy użyciu wątki na to jest przesada . Możesz upewnić się, że plik jest zapisany poprawnie, po prostu robi to w pętli aż udany zapisu zrobiono:

def saveToFile(obj, filename): 
    file = open(filename, 'w') 
    cPickle.dump(obj, file) 
    file.close() 
    return True 

done = False 
while not done: 
    try: 
     done = saveToFile(obj, 'file') 
    except KeyboardInterrupt: 
     print 'retry' 
     continue 

Answer 4

Poniżej znajduje się kierownik kontekst że przywiązuje obsługi sygnału dla SIGINT. Jeśli wywoływana jest obsługa sygnału menedżera kontekstu, sygnał jest opóźniany przez przekazanie sygnału do oryginalnej procedury obsługi tylko po wyjściu menedżera kontekstów.

import signal 
import logging 

class DelayedKeyboardInterrupt(object): 
    def __enter__(self): 
     self.signal_received = False 
     self.old_handler = signal.signal(signal.SIGINT, self.handler) 

    def handler(self, sig, frame): 
     self.signal_received = (sig, frame) 
     logging.debug('SIGINT received. Delaying KeyboardInterrupt.') 

    def __exit__(self, type, value, traceback): 
     signal.signal(signal.SIGINT, self.old_handler) 
     if self.signal_received: 
      self.old_handler(*self.signal_received) 

with DelayedKeyboardInterrupt(): 
    # stuff here will not be interrupted by SIGINT 
    critical_code()