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

Question

我需要实现这样的事情：

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

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

但是我需要不仅在每次新连接之后调用该restore()函数，而且在 p 死后立即调用该函数。

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

换句话说，我需要阻止我的程序，直到这两个条件之一为真：“p dies”或“new connection”。

我知道我可以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.