当父进程死亡时,如何杀死使用 subprocess.check_output() 创建的 python 子进程?
- 2024-10-17 08:47:00
- admin 原创
- 69
问题描述:
我在 Linux 机器上运行一个 Python 脚本,该脚本使用 subprocess.check_output() 创建子进程,如下所示:
subprocess.check_output(["ls", "-l"], stderr=subprocess.STDOUT)
问题是,即使父进程终止,子进程仍然在运行。当父进程终止时,我有什么方法可以终止子进程吗?
解决方案 1:
您的问题在于使用subprocess.check_output
- 您是对的,您无法使用该接口获取子 PID。请改用 Popen:
proc = subprocess.Popen(["ls", "-l"], stdout=PIPE, stderr=PIPE)
# Here you can get the PID
global child_pid
child_pid = proc.pid
# Now we can wait for the child to complete
(output, error) = proc.communicate()
if error:
print "error:", error
print "output:", output
为了确保你在退出时杀死孩子:
import os
import signal
def kill_child():
if child_pid is None:
pass
else:
os.kill(child_pid, signal.SIGTERM)
import atexit
atexit.register(kill_child)
解决方案 2:
是的,您可以通过两种方法实现这一点。它们都要求您使用Popen
而不是check_output
。第一种方法更简单,使用 try..finally,如下所示:
from contextlib import contextmanager
@contextmanager
def run_and_terminate_process(*args, **kwargs):
try:
p = subprocess.Popen(*args, **kwargs)
yield p
finally:
p.terminate() # send sigterm, or ...
p.kill() # send sigkill
def main():
with run_and_terminate_process(args) as running_proc:
# Your code here, such as running_proc.stdout.readline()
这将捕获 sigint(键盘中断)和 sigterm,但不会捕获 sigkill(如果您使用 -9 终止脚本)。
另一种方法稍微复杂一些,使用 ctypes 的 prctl PR_SET_PDEATHSIG。一旦父进程因任何原因退出(甚至是 sigkill),系统就会向子进程发送信号。
import signal
import ctypes
libc = ctypes.CDLL("libc.so.6")
def set_pdeathsig(sig = signal.SIGTERM):
def callable():
return libc.prctl(1, sig)
return callable
p = subprocess.Popen(args, preexec_fn = set_pdeathsig(signal.SIGTERM))
解决方案 3:
不知道具体情况,但最好的方法仍然是用信号捕获错误(甚至可能是所有错误)并终止那里任何剩余的进程。
import signal
import sys
import subprocess
import os
def signal_handler(signal, frame):
sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)
a = subprocess.check_output(["ls", "-l"], stderr=subprocess.STDOUT)
while 1:
pass # Press Ctrl-C (breaks the application and is catched by signal_handler()
这只是一个模型,您需要捕捉的不仅仅是 SIGINT,但这个想法可能会让您开始,并且您仍然需要以某种方式检查生成的进程。
http://docs.python.org/2/library/os.html#os.kill
http://docs.python.org/2/library/subprocess.html#subprocess.Popen.pid
http://docs。 python.org/2/library/subprocess.html#subprocess.Popen.kill
我建议重写一个个性化版本,check_output
因为我刚刚意识到 check_output 实际上只是为了简单的调试等,因为在执行过程中你不能与它进行太多交互。
重写check_output:
from subprocess import Popen, PIPE, STDOUT
from time import sleep, time
def checkOutput(cmd):
a = Popen('ls -l', shell=True, stdin=PIPE, stdout=PIPE, stderr=STDOUT)
print(a.pid)
start = time()
while a.poll() == None or time()-start <= 30: #30 sec grace period
sleep(0.25)
if a.poll() == None:
print('Still running, killing')
a.kill()
else:
print('exit code:',a.poll())
output = a.stdout.read()
a.stdout.close()
a.stdin.close()
return output
然后用它做任何你想做的事情,也许将活动执行存储在临时变量中,并在退出时使用信号或其他方式拦截主循环的错误/关闭。
最后,您仍然需要在主应用程序中捕获终止,以便安全地杀死任何子进程,解决这个问题的最佳方法是使用try & except
或signal
。
解决方案 4:
从 Python 3.2 开始,有一个非常简单的方法可以做到这一点:
from subprocess import Popen
with Popen(["sleep", "60"]) as process:
print(f"Just launched server with PID {process.pid}")
我认为这对大多数用例来说都是最好的,因为它简单且可移植,并且避免了对全局状态的任何依赖。
如果这个解决方案不够强大,那么我建议查看关于这个问题或Python 的其他答案和讨论:当父进程死亡时如何杀死子进程?,因为有很多巧妙的方法可以解决这个问题,这些方法在可移植性、弹性和简单性方面提供不同的权衡。
- 2024年20款好用的项目管理软件推荐,项目管理提效的20个工具和技巧
- 2024年开源项目管理软件有哪些?推荐5款好用的项目管理工具
- 项目管理软件有哪些?推荐7款超好用的项目管理工具
- 项目管理软件哪个最好用?盘点推荐5款好用的项目管理工具
- 项目管理软件有哪些最好用?推荐6款好用的项目管理工具
- 项目管理软件有哪些,盘点推荐国内外超好用的7款项目管理工具
- 2024项目管理软件排行榜(10类常用的项目管理工具全推荐)
- 项目管理软件排行榜:2024年项目经理必备5款开源项目管理软件汇总
- 2024年常用的项目管理软件有哪些?推荐这10款国内外好用的项目管理工具
- 项目管理必备:盘点2024年13款好用的项目管理软件