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

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

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

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

subprocess.CREATE_NEW_PROCESS_GROUPA 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竟然没有那么多参数，而且没有返回值，简陋啊。不过正解决了我的问题。

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

[[email protected] 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)