当父进程死亡时,如何杀死使用 subprocess.check_output() 创建的 python 子进程?

2024-10-17 08:47:00
admin
原创
68
摘要:问题描述:我在 Linux 机器上运行一个 Python 脚本,该脚本使用 subprocess.check_output() 创建子进程,如下所示:subprocess.check_output(["ls", "-l"], stderr...

问题描述:

我在 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 & exceptsignal

解决方案 4:

从 Python 3.2 开始,有一个非常简单的方法可以做到这一点:

from subprocess import Popen

with Popen(["sleep", "60"]) as process:
    print(f"Just launched server with PID {process.pid}")

我认为这对大多数用例来说都是最好的,因为它简单且可移植,并且避免了对全局状态的任何依赖。

如果这个解决方案不够强大,那么我建议查看关于这个问题或Python 的其他答案和讨论:当父进程死亡时如何杀死子进程?,因为有很多巧妙的方法可以解决这个问题,这些方法在可移植性、弹性和简单性方面提供不同的权衡。

相关推荐
  为什么项目管理通常仍然耗时且低效?您是否还在反复更新电子表格、淹没在便利贴中并参加每周更新会议?这确实是耗费时间和精力。借助软件工具的帮助,您可以一目了然地全面了解您的项目。如今,国内外有足够多优秀的项目管理软件可以帮助您掌控每个项目。什么是项目管理软件?项目管理软件是广泛行业用于项目规划、资源分配和调度的软件。它使项...
项目管理软件   601  
  华为IPD与传统研发模式的8大差异在快速变化的商业环境中,产品研发模式的选择直接决定了企业的市场响应速度和竞争力。华为作为全球领先的通信技术解决方案供应商,其成功在很大程度上得益于对产品研发模式的持续创新。华为引入并深度定制的集成产品开发(IPD)体系,相较于传统的研发模式,展现出了显著的差异和优势。本文将详细探讨华为...
IPD流程是谁发明的   7  
  如何通过IPD流程缩短产品上市时间?在快速变化的市场环境中,产品上市时间成为企业竞争力的关键因素之一。集成产品开发(IPD, Integrated Product Development)作为一种先进的产品研发管理方法,通过其结构化的流程设计和跨部门协作机制,显著缩短了产品上市时间,提高了市场响应速度。本文将深入探讨如...
华为IPD流程   9  
  在项目管理领域,IPD(Integrated Product Development,集成产品开发)流程图是连接创意、设计与市场成功的桥梁。它不仅是一个视觉工具,更是一种战略思维方式的体现,帮助团队高效协同,确保产品按时、按质、按量推向市场。尽管IPD流程图可能初看之下显得错综复杂,但只需掌握几个关键点,你便能轻松驾驭...
IPD开发流程管理   8  
  在项目管理领域,集成产品开发(IPD)流程被视为提升产品上市速度、增强团队协作与创新能力的重要工具。然而,尽管IPD流程拥有诸多优势,其实施过程中仍可能遭遇多种挑战,导致项目失败。本文旨在深入探讨八个常见的IPD流程失败原因,并提出相应的解决方法,以帮助项目管理者规避风险,确保项目成功。缺乏明确的项目目标与战略对齐IP...
IPD流程图   8  
热门文章
项目管理软件有哪些?
云禅道AD
禅道项目管理软件

云端的项目管理软件

尊享禅道项目软件收费版功能

无需维护,随时随地协同办公

内置subversion和git源码管理

每天备份,随时转为私有部署

免费试用