Как правильно завершить работу потоков по KeyboardInterrupt?

Question

[Андрей Горбик]

Хочу по KeyboardInterrupt просто выходить из программы, однако
не понимаю, где обрабатывать это исключение.

from threading import Thread
from time import sleep

def worker(i):
    while True:
        print("worker {}".format(i))
        sleep(i)

threads = [Thread(target=worker, args=(i,)) for i in range(1, 4)]
for t in threads:
    t.start()

for t in threads:
    t.join()

Answer 1

[Сергей Тихонов]

https://docs.python.org/3/library/threading.html

A thread can be flagged as a “daemon thread”. The significance of this flag is that the entire Python program exits when only daemon threads are left. The initial value is inherited from the creating thread. The flag can be set through the daemon property or the daemon constructor argument.

Если обработка KeyboardInterrupt сама по себе не нужна, а хотите чтобы процесс по Ctrl+C завершался - объявите все потоки демонами.

Comments

[Андрей Горбик]

Для данной задачи этот способ наверное действительно хорош, но я хотел бы разобраться с обработкой исключений при использовании потоков (в общем случае).

[Сергей Тихонов]

В общем случае

stopped = False
while not stopped:
    try:
        do_work()
    except KeyboardInterrupt:
        stopped = True

Один поток обрабатывать Ctrl+C, триггерит флаг остановки. Остальные потоки замечают этот флаг и тоже останавливаются. Не нашел только сходу в документации, в какой поток прилетает KeyboardInterrupt.

[Андрей Горбик]

Сергей Тихонов, вы не могли бы объяснить, как предложенный Вами способ поможет, если KeyboardInterrupt прилетит в главный поток. Сколько я не пытался сделать код рабочим, так и не смог:

from threading import Thread
from time import sleep

is_stopped = False

def worker(i):
    global is_stopped

    while not is_stopped:
        try:
            print("worker {}".format(i))
            sleep(i)
        except KeyboardInterrupt:
            is_stopped = True

threads = [Thread(target=worker, args=(i,)) for i in range(1, 4)]
for t in threads:
    t.start()

for t in threads:
    t.join()

При запуске и попытке остановить через ctrl + c ничего не получается.

[Сергей Тихонов]

Андрей Горбик, главный поток у Вас не ловит исключения, пока на join сидит. А должен.

[Андрей Горбик]

Сергей Тихонов, все равно не выходит. Ладно, сделал, как вы советовали через демона. Так работает, спасибо.

from threading import Thread
from time import sleep

def worker(i):
    while True:
        print("worker {}".format(i))
        sleep(i)
    
threads = [Thread(target=worker, args=(i,), daemon=True) for i in range(1, 4)]

for t in threads:
    t.start()

try:
    for t in threads:
        t.join()
except KeyboardInterrupt:
    pass

[Сергей Тихонов]

Андрей Горбик, ну так основной поток тоже должен стоп-флаг выставлять:

from threading import Thread
from time import sleep, time


active = True

def worker(i):
    global active
    while active:
        try:
            print(f"{time()}: worker {i} is active")
            sleep(i)
        except KeyboardInterrupt:
            active = False
            print(f'{time()} worker {i} received SIGINT')


threads = [Thread(target=worker, args=(i,)) for i in range(1, 4)]

for t in threads:
    t.start()

try:
    for t in threads:
        t.join()
except KeyboardInterrupt:
    print(f"{time()}: main thread received sigint")
    active = False
    for t in threads:
        t.join()
print(f"{time()}: all threads gone")

1541616371.7186885: worker 1 is active
1541616371.718856: worker 2 is active
1541616371.7192252: worker 3 is active
1541616372.7196927: worker 1 is active
1541616372.770136: main thread received sigint
1541616374.7199564: all threads gone

[Андрей Горбик]

Сергей Тихонов, этого я не учел.. Спасибо большое!

[Андрей Горбик]

Сергей Тихонов, заметил, что никогда не печатается строчка "worker {i} received SIGNINT". Мелькнула догадка, что исключение не ловится в воркерах. Убрал обработку в них и все также работает. Видимо, KeyboardInterrupt всегда прилетает в главный поток.