关闭交互式 python 会话时结束非守护线程

Question

请考虑下面的代码：

#! /usr/bin/env python3

import threading
import time

class MyThread(threading.Thread):
    def __init__(self):
        super().__init__()
        self._quit_flag = False
    def run(self):
        while not self._quit_flag:
            print("Thread is alive!")
            time.sleep(0.500)
    def request_quit(self):
        self._quit_flag = True

mt = MyThread()
mt.start()

将其保存为“test.py”并运行“python3 -i test.py”后，我得到了一个交互式会话，线程定期打印“线程活着！”消息。

当我按下 control-D 或运行 exit() 时，python3 进程不会终止，因为线程仍在运行。我想解决这个问题。

但是，我不想让线程成为守护线程——我希望能够正确地结束/加入线程，因为在我正在解决的现实案例中，我想很好地终止网络连接。

有没有办法做到这一点？例如，在 REPL 循环终止之后，就在主线程退出之前执行的某个钩子可用吗？

Answer 1

好的，我自己找到了一种方法。

查看 Python 源代码，事实证明，在关闭时，Python 会在完成进程关闭之前尝试 join() 所有非守护线程。哪种是有意义的，除了如果有记录的方式来通知这些线程，这将更加有用。没有。

但是，是可以重新实现一个自己的从 Thread 派生的类的 join() 方法，这可以作为通知子线程它应该关闭自己的一种方式下：

class MyThread(threading.Thread):
    def __init__(self):
        super().__init__()
        self._quit_flag = False
    def __del__(self):
        print("Bye bye!")
    def run(self):
        while not self._quit_flag:
            print("Thread is alive!",  threading.active_count(), threading.main_thread().is_alive())
            for t in threading.enumerate():
                print(t.is_alive())
            time.sleep(0.500)
    def request_quit(self):
        self._quit_flag = True

    def join(self):
        self.request_quit()
        super().join()

但是，这种解决问题的方法几乎是一种 hack，它忽略了“线程”模块文档 (https://docs.python.org/3/library/threading.html) 中的以下段落：

Thread 类表示一个在单独的 控制线程。有两种方法可以指定活动：通过 将可调用对象传递给构造函数，或者通过覆盖 子类中的 run() 方法。 没有其他方法（除了 构造函数）应该在子类中被覆盖。换句话说，只有 覆盖该类的 __init__() 和 run() 方法。

不过，它确实工作得很好。

Answer 2

我不确定这是否可以接受，但我能看到您的问题的唯一方法是从您的脚本启动交互式控制台，而不是使用 -i 标志。然后，在您终止交互式会话后，您将能够继续执行程序以退出：

import threading
import time
from code import InteractiveConsole

def exit_gracefully():
    print("exiting gracefully now")
    global mt
    mt.request_quit()
    mt.join()


class MyThread(threading.Thread):
    def __init__(self):
        super().__init__()
        self._quit_flag = False
    def run(self):
        while not self._quit_flag:
            print("Thread is alive!")
            time.sleep(0.500)
    def request_quit(self):
        self._quit_flag = True

mt = MyThread()
mt.start()
InteractiveConsole(locals=locals()).interact()
exit_gracefully()

这个你可以在没有-i 标志的情况下执行，仅作为

python3 test.py

它仍然会像 python -i 一样为您提供交互式控制台，但现在脚本在交互式 shell 退出后执行 exit_gracefully()。

Answer 3

我认为您可以这样做的方法是使用atexit 注册一个清理函数并将您的线程设置为守护进程，例如：

#! /usr/bin/env python3

import threading
import time
import atexit

class MyThread(threading.Thread):
    def __init__(self):
        super().__init__()

        # this ensures that atexit gets called.
        self.daemon = True
        self._quit_flag = False

    def run(self):
        while not self._quit_flag:
            print("Thread is alive!")
            time.sleep(0.500)
        print("cleanup can also go here")

    def request_quit(self):
        print("cleanup can go here")
        self._quit_flag = True


mt = MyThread()
def cleanup():
    mt.request_quit()
    mt.join()
    print("even more cleanup here, so many options!")

atexit.register(cleanup)
mt.start()

Answer 4

只需在后台创建一个新的关闭线程：

import threading
import time

class MyThread(threading.Thread):
    def __init__(self):
        super().__init__()
        self._quit_flag = False
    def run(self):
        while not self._quit_flag:
            print("Thread is alive!")
            time.sleep(0.500)
        print("shutting down.")
    def request_quit(self):
        self._quit_flag = True

mt = MyThread()
mt.start()

# Shutdown thread
threading.Thread(
    target=lambda: threading.main_thread().join() or mt.request_quit()
    ).start()