Как правильно обрабатывать SIGINT для закрытия файлов/соединений

Я бы предпочел поймать исключение KeyboardInterrupt в основном потоке. KeyboardInterrupt является результатом обработчика SIGINT по умолчанию для python. Обработчик исключения KeyboardInterrupt является гораздо более безопасным/дружественным контекстом, чем тот, в котором вы находитесь при непосредственном перехвате SIGINT.

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        cleanup()

РЕДАКТИРОВАТЬ: Вот как разделить переменные (состояние) между двумя методами:

Процедурный:

import sys, time

class SharedState:
    def __init__(self):
        self.var0 = 42
        self.var1 = 'string'

# method 1
shared_variable = 'woof woof'

# method 2: avoiding global declarations in functions
shared_state = SharedState()

def main():
    # In order to write a global variable you need a global
    # declaration otherwise the assignment would create a
    # local variable

    global shared_variable
    shared_variable = 5

    shared_state.var0 = 10

    time.sleep(10)

def cleanup():
    print shared_variable
    print shared_state.var0
    sys.exit(1)

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        cleanup()

Объектно-ориентированный (мое предпочтение):

import sys, time

# method 3: object oriented programming
class Program:
    def __init__(self):
        self.var0 = 42
        self.var1 = 'string'

    def main(self):
        self.var0 = 5
        self.var1 = 'woof woof'
        time.sleep(10)

    def cleanup(self):
        # both main and cleanup can access the member
        # variables of this class
        print self.var0
        print self.var1
        sys.exit(1)

    def execute(self):
        try:
            self.main()
        except KeyboardInterrupt:
            self.cleanup()

if __name__ == '__main__':
    Program().execute()

Я предлагаю использовать библиотеку signal для обработки сигналов. Сигналы не являются исключением и являются частью инфраструктуры Inter Process Communication (IPC) операционной системы. Система.

Сигналы могут помочь вам взаимодействовать с вашей программой, например, перезагрузить файл конфигурации, закрыть обработчик файла журнала во время ротации журнала и так далее. Большая часть процесса демона, такого как apache, дозирует его.

В сценариях оболочки есть команда trap для обработки сигналов и выполнения соответствующих действий на основе полученных сигналов.

Обычно python автоматически закрывает все обработчики файлов и соединение с базой данных во время выхода. Но на всякий случай у нас может быть функция для неявной обработки.

Код ниже перехватывает SIGINT и правильно закрывает файлы.

import signal
import sys

die = False

def handler(signum, frame):
    global die
    print('Got SIGINT.')
    die = True

def closeFile(fh):
    fh.flush()
    fh.close()

signal.signal(signal.SIGINT, handler)

fh = open('/tmp/a.txt', 'w')

while True:
    data = input('> ')

    if data == 'q':
        closeFile(fh)
        break
    else:
        fh.write(data + '\n')

    if die:
        closeFile(fh)
        print('Completed cleanup.. ')
        sys.exit()