确保子进程在退出 Python 程序时死亡

2022-01-18 00:00:00 python subprocess kill zombie-process

问题描述

有没有办法确保所有创建的子进程在 Python 程序退出时都死了?我所说的子进程是指那些使用 subprocess.Popen() 创建的.

Is there a way to ensure all created subprocess are dead at exit time of a Python program? By subprocess I mean those created with subprocess.Popen().

如果不是,我应该遍历所有发出的杀戮然后杀掉 -9 吗?有什么清洁的吗?

If not, should I iterate over all of the issuing kills and then kills -9? anything cleaner?


解决方案

你可以使用 atexit 为此,并注册任何清理任务以在程序退出时运行.

You can use atexit for this, and register any clean up tasks to be run when your program exits.

atexit.register(func[, *args[, **kargs]])

在您的清理过程中,您还可以实现自己的等待,并在您希望的超时发生时将其终止.

In your cleanup process, you can also implement your own wait, and kill it when a your desired timeout occurs.

>>> import atexit
>>> import sys
>>> import time
>>> 
>>> 
>>>
>>> def cleanup():
...     timeout_sec = 5
...     for p in all_processes: # list of your processes
...         p_sec = 0
...         for second in range(timeout_sec):
...             if p.poll() == None:
...                 time.sleep(1)
...                 p_sec += 1
...         if p_sec >= timeout_sec:
...             p.kill() # supported from python 2.6
...     print 'cleaned up!'
...
>>>
>>> atexit.register(cleanup)
>>>
>>> sys.exit()
cleaned up!

注意 -- 如果该进程(父进程)被杀死,注册的函数将不会运行.

Note -- Registered functions won't be run if this process (parent process) is killed.

python >= 2.6 不再需要以下windows方法

这是一种在 Windows 中终止进程的方法.您的 Popen 对象具有 pid 属性,因此您可以通过 success = win_kill(p.pid) 调用它(需要 pywin32 已安装):

Here's a way to kill a process in windows. Your Popen object has a pid attribute, so you can just call it by success = win_kill(p.pid) (Needs pywin32 installed):

    def win_kill(pid):
        '''kill a process by specified PID in windows'''
        import win32api
        import win32con

        hProc = None
        try:
            hProc = win32api.OpenProcess(win32con.PROCESS_TERMINATE, 0, pid)
            win32api.TerminateProcess(hProc, 0)
        except Exception:
            return False
        finally:
            if hProc != None:
                hProc.Close()

        return True

相关文章