python subprocess 杀掉全部派生的子进程

 下面就是今天下午的研究成果。

   发布系统需要响应用户的中断请求，需要在GET方法中杀掉由subprocess派生的子进程，刚开始直接用os.kill 发现子进程的子进程无法kill，谷歌了一些，发现kill可以干掉进程组，于是测试，但是默认情况下，subprocess派生的进程组和主程序，也就是我的web.py进程是在一个进程组里的，这要是kill了，那就调的了。

    继续翻google，看subprocess的document时发现这个变量：

subprocess.CREATE_NEW_PROCESS_GROUP

A Popen creationflags parameter to specify that a new process group will be created. This flag is necessary for using os.kill() on the subprocess.

This flag is ignored if CREATE_NEW_CONSOLE is specified.

比较高兴，以为能解决问题了，结果测试半天，才了解这玩意是only windows的，我去啊，不过想到了，win能做到的，linux肯定也可以，于是定位到

preexec_fn

 

又是一通google，不是对象吗，弄了个setpgid(0,0) 测试了，子进程还是和主调进程属于同一个进程组，后来灵机一动：

 

preexec_fn = os.setpgrp

 

这样竟然解决了新生成进程组的问题。

继续努力，后面遇到的就是僵死进程的问题了，os.waitpid了一下就解决了。

刚开始waitpid的时候，还在linxu上man了半天，看着linxu手册里的参数，还是不放心啊，结果python里的os.waitpid竟然没有那么多参数，而且没有返回值，简陋啊。不过正解决了我的问题。

 

下面是今天的完全测试代码。

 

 

 

[liufeng@1.2.3.4 kill-subprocess]$ cat sub-process.py 

import subprocess

import os

import time

 

 

def my_func():

#派生两个子进程，子进程里又派生几个sleep的孙子进程，主要是为了测试kill进程组。

                run_str2 = '/bin/sh test.sh'

                run_str = '/bin/sh test_quick.sh'

                cmd2 = run_str.split()

                cmd = run_str.split()

#测试了一些个preexec_fn的值，最终发现能用的，对python的对象的概念还是不理解啊，新手，新手。

                #p = subprocess.Popen(cmd, stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = False, creationflags = subprocess.CREATE_NEW_PROCESS_GROUP)

                #p = subprocess.Popen(cmd, stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = False, creationflags = 0)

                p = subprocess.Popen(cmd, stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = False, preexec_fn = os.setpgrp )

                p2 = subprocess.Popen(cmd2, stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = False, preexec_fn = os.setpgrp )

                #@p = subprocess.Popen(cmd, stdin = subprocess.PIPE, stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = False, preexec_fn = os.setpgid(0, 0) )

 

                pid = p.pid

                pgid = os.getpgid(pid)

                print "pid: %d\n" %pid

                print "pgid: %d\n" %pgid

                return pid

 

pid = my_func()

#p.wait()

print "now , sleep 2s ,then , os.kill gpid %d" % pid

time.sleep(20)

 

a = os.kill(-pid, 9)

print "kill,return:"

print a

 

# kill的时候，我测试了kill 没有权限的root进程，会报错：权限不允许

# 测试了kill p p2 都可以kill

#a = os.kill(2445, 9)

#print "kill root process 2445 ,return:"

#print a

#p.wait()

#os.waitpid(pgid, 0)

# 2445 is a root process

#os.waitpid(2445, 0)

#os.waitpid(p2.pid, 0)

os.waitpid(pid, 0)

print "waitpid,return:"

print a

time.sleep(22)

 

print "done..."

 

#p.terminate()

#p.kill()

#p.wait()

#

#time.sleep(40)

#os.kill(pid, 9)