subprocess.popen.kill杀死所有子进程

py3study

发布于 2020-02-25 12:03:51

5.2K0

发布于 2020-02-25 12:03:51

文章被收录于专栏：python3python3

一、使用subprocess模块

使用subprocess模块可创建子进程。

subprocess. Popen ( args , bufsize=0 , executable=None , stdin=None , stdout=None , stderr=None , preexec_fn=None , close_fds=False , shell=False , cwd=None , env=None , universal_newlines=False , startupinfo=None , creationflags=0 )

subprocess.Popen(['google-chrome',"http://www.baidu.com"])可打开chrome并显示百度网页。

subprocess.Popen.pid可显示子进程的id，但这只是打开的shell的id，如果此时使用pid.kill是无法杀掉所有子进程的。

杀掉所有子进程，代码：

import sys, os, subprocess, time, signal
p = subprocess.Popen(['google-chrome',"http://www.baidu.com"], close_fds=True, preexec_fn = os.setsid)
print p.pid  
time.sleep(5)
# p.kill() #无法杀掉所有子进程，只能杀掉子shell的进程

# p.terminate()  #无法杀掉所有子进程
os.killpg( p.pid,signal.SIGUSR1)

time.sleep(3)

注意：

close_fds=True，此时除了文件描述符为0 , 1 and 2，其他子进程都要被杀掉。( Linux中所有的进程都是进程0的子进程。

pid=1的是init，内核完成之后启动的第一个进程，然后init根据/etc/inittab的内容再去启动其它进程。)

os.setsid()：使独立于终端的进程（不响应sigint，sighup等），使脱离终端。

SIGUSR1：用户自定义信号

os.killpg( p.pid,signal.SIGUSR1)：发送SIGUSR1信号到组进程p.pid

二、封装subprocess 函数

最近使用python 调用ansible命令，来监测系统相关服务。发现到达指定时间，ansible进程没有杀掉。导致服务器卡死。

后来尝试使用os.killpg ，发现可以杀掉ansible 子进程。

完整代码如下：

#!/usr/bin/env python3
# coding: utf-8

import os
import time
import signal
import subprocess

def execute_linux2(cmd, timeout=10, skip=False):
    """
    执行linux命令,返回list
    :param cmd: linux命令
    :param timeout: 超时时间,生产环境, 特别卡, 因此要3秒
    :param skip: 是否跳过超时限制
    :return: list
    """
    p = subprocess.Popen(cmd, stderr=subprocess.STDOUT, stdout=subprocess.PIPE,shell=True,close_fds=True,preexec_fn=os.setsid)

    t_beginning = time.time()  # 开始时间
    while True:
        if p.poll() is not None:
            break
        seconds_passed = time.time() - t_beginning
        if not skip:
            if seconds_passed > timeout:
                # p.terminate()
                # p.kill()
                # raise TimeoutError(cmd, timeout)
                print('错误, 命令: {},本地执行超时!'.format(cmd))
                # 当shell=True时，只有os.killpg才能kill子进程
                try:
                    # time.sleep(1)
                    os.killpg(p.pid, signal.SIGUSR1)
                except Exception as e:
                    pass
                return False

    result = p.stdout.readlines()  # 结果输出列表
    return result

本文参考链接：

https://blog.csdn.net/heihuifeng/article/details/6543414

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2020-02-22 ，如有侵权请联系 cloudcommunity@tencent.com 删除

DevOps 解决方案