守护线程启动软件不会死

Question

我正在尝试编写一个小脚本，它将使用 plink.exe（来自同一文件夹）创建一个 ssh 隧道（在 Windows 上）。 我基本上是使用os.system 来启动命令：

import time
import threading
from os.path import join, dirname, realpath
pc_tunnel_command = '-ssh -batch -pw xxxx -N -L 1234:host1:5678 user@host2'

if __name__ == '__main__':
    t = threading.Thread(target = os.system, \
                         args = (join(dirname(realpath(__file__)), 'plink.exe ') + \
                                      pc_tunnel_command,))
    t.daemon = True
    t.start()
    #without this line it will die. I guess that plink doesn't have enough time to start.
    time.sleep(5) 
    print 'Should die now'

但是，线程（和 plink.exe）似乎继续运行。为什么会这样？有什么方法可以强制线程关闭？启动 plink 的更好方法？

我希望 plink.exe 在我的程序结束时死掉。使用守护线程是我让隧道在后台运行的计划，然后在我的主代码退出时死掉。

顺便说一句 - subprocess.call 也会发生同样的事情。

Answer 1

您可以使用atexit 和signal 模块注册回调，当您的程序正常退出或接收到SIGTERM 时，它们将显式终止进程：

import sys
import time
import atexit
import signal
import subprocess
from functools import partial
from os.path import join, dirname, realpath

pc_tunnel_command = '-ssh -batch -pw xxxx -N -L 1234:host1:5678 user@host2'


def handle_exit(p, *args):
    print("killing it")
    p.terminate()
    sys.exit(0)

if __name__ == '__main__':
    p = subprocess.Popen(join(dirname(realpath(__file__)), 'plink.exe ') + pc_tunnel_command, shell=True)
    func = partial(handle_exit, p)
    signal.signal(signal.SIGTERM, func)
    atexit.register(func)
    print 'Should die now'

关于您所描述的行为的一件奇怪的事情是，我希望您的程序在您的 sleep 调用之后退出，但让 plink 在后台运行，而不是让您的程序挂起直到 os.system通话完成。这就是我在 Linux 上看到的行为，至少。无论如何，显式终止子进程应该可以为您解决问题。

Answer 2

os.system 在子进程退出之前不会返回。 subprocess.call 也是如此。这就是为什么你的线程坐在那里，等待 plink 完成。您可能可以使用subprocess.Popen 异步启动进程然后退出。无论如何，您创建的附加线程是不必要的。