Как запустить последнюю функцию перед тем, как ее убьют в Python?

Question

Есть ли какой-нибудь способ запустить последнюю команду перед тем, как запущенный скрипт Python будет остановлен из-за прерывания каким-либо другим скриптом, клавиатуры и т.д.

Answer 1

import time

try:
    time.sleep(10)
finally:
    print "clean up"

clean up
Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
KeyboardInterrupt

Если вам нужно перехватить другие прерывания уровня операционной системы, посмотрите на сигнальный модуль:

http://docs.python.org/library/signal.html

Пример сигнала

from signal import *
import sys, time

def clean(*args):
    print "clean me"
    sys.exit(0)

for sig in (SIGABRT, SIGBREAK, SIGILL, SIGINT, SIGSEGV, SIGTERM):
    signal(sig, clean)

time.sleep(10)

Answer 2

Вы могли бы использовать atexit модуль.С его помощью вы можете зарегистрировать функцию, которая будет вызвана при завершении программы.Пример отсюда: http://docs.python.org/library/atexit.html

try:
    _count = int(open("/tmp/counter").read())
except IOError:
    _count = 0

def incrcounter(n):
    global _count
    _count = _count + n

def savecounter():
    open("/tmp/counter", "w").write("%d" % _count)

import atexit
atexit.register(savecounter)

Вы также можете передать позиционные параметры и ключевое слово функции, которую вы хотите вызвать при завершении программы.

Обратите внимание, что в документах указано несколько обстоятельств, при которых ваш обработчик не будет вызван:

Примечание:Функции, зарегистрированные с помощью этого модуля, не вызываются, когда программа завершает работу из-за сигнала, не обработанного Python, при обнаружении фатальной внутренней ошибки Python или когда os._exit() называется.

Таким образом, вы можете также зарегистрировать обработчик сигналов.

Answer 3

import signal
import sys
import time

def cleanup(*args):
    print 'Exiting'
    sys.exit(0)

signal.signal(signal.SIGINT, cleanup)
signal.signal(signal.SIGTERM, cleanup)
while True:
    time.sleep(60)  # less busy loop

Answer 4

Приношу извинения "Неизвестному" за то, что принял их ответ и исправил его, как если бы это был мой собственный ответ, но мои правки были отклонены.

Утвержденный ответ содержит ошибку, которая приведет к segfault.

Вы не можете использовать sys.exit() в обработчике сигналов, но вы можете использовать os._exit, чтобы он стал:

from signal import *
import os, time

def clean(*args):
    print "clean me"
    os._exit(0)

for sig in (SIGABRT, SIGINT, SIGTERM):
    signal(sig, clean)

time.sleep(10)

SIGBREAK может быть использован, если целевой платформой является Windows.

В зависимости от варианта использования и необходимости очистки в случае неустранимых ошибок - вы можете добавить SIGSEGV и SIGILL, но обычно это не рекомендуется, поскольку состояние программы может быть таким, что вы создадите бесконечный цикл.

Answer 5

Используйте модуль atexit для регистрации функции, которая будет вызвана в конце.

import atexit
atexit.register(some_function)