Python：等待“子进程终止”或“新连接”

Question

我需要实现这样的事情：

while True:
    conn,addr = socket.accept()
    restore()
    p.kill()

    #...
    p = subprocess.Popen(...)

但是我不仅需要在每个新连接之后都调用restore()函数，而且还必须在p终止时调用该函数。

我可以通过执行p.wait()来“阻止”等待p死亡的程序，但是同时，我也想阻止等待新连接的程序。

换句话说，我需要阻塞程序，直到这两个条件之一成立：“ p dies”或“新连接”。

我知道可以使用select等待两个文件描述符，但是在这种情况下我不知道该怎么做。

谢谢

Answer 1

这是我最终使用的解决方案：

def wait_socket_and_process(s,process):
    while True:
        rr,_,_ = select.select([s],[],[],0.1)
        if len(rr)>0:
            conn, addr = rr[0].accept()
            process.kill()
            process.poll()
            #handle process termination
            print "* subprocess killed"
            return conn, addr

        elif (process.poll() != None):
            #handle process termination
            print "* subprocess terminated"


s = socket.#...
process = subprocess.Popen(...)
wait_socket_and_process(s,process)
#handle new connection...

正如其他人所建议的那样，使用getenv库可能会有更清洁/更通用的方法来实现此目的。