Python：如何窥视 pty 对象以避免阻塞？

Question

我正在使用pty 读取非阻塞进程的标准输出，如下所示：

import os
import pty
import subprocess

master, slave = pty.openpty()

p = subprocess.Popen(cmd, stdout = slave)

stdout = os.fdopen(master)
while True:
    if p.poll() != None:
        break

    print stdout.readline() 

stdout.close()

除了while-loop 偶尔会阻塞之外，一切正常。这是因为print stdout.readline() 行正在等待从stdout 读取某些内容。但是如果程序已经终止，我上面的小脚本就会永远挂起。

我的问题是：有没有办法窥视stdout 对象并检查是否有数据可供读取？如果不是这种情况，它应该继续通过while-loop，它会发现进程实际上已经终止并中断循环。

Answer 1

是的，使用select module's poll：

import select
q = select.poll()
q.register(stdout,select.POLLIN)

同时使用：

l = q.poll(0)
if not l:
    pass # no input
else:
    pass # there is some input

Answer 2

select.poll() 答案非常简洁，但不适用于 Windows。以下解决方案是一种替代方案。它不允许您查看标准输出，但提供了 readline() 的非阻塞替代方案，并且基于 this answer:

from subprocess import Popen, PIPE
from threading import Thread
def process_output(myprocess): #output-consuming thread
    nextline = None
    buf = ''
    while True:
        #--- extract line using read(1)
        out = myprocess.stdout.read(1)
        if out == '' and myprocess.poll() != None: break
        if out != '':
            buf += out
            if out == '\n':
                nextline = buf
                buf = ''
        if not nextline: continue
        line = nextline
        nextline = None

        #--- do whatever you want with line here
        print 'Line is:', line
    myprocess.stdout.close()

myprocess = Popen('myprogram.exe', stdout=PIPE) #output-producing process
p1 = Thread(target=process_output, args=(myprocess,)) #output-consuming thread
p1.daemon = True
p1.start()

#--- do whatever here and then kill process and thread if needed
if myprocess.poll() == None: #kill process; will automatically stop thread
    myprocess.kill()
    myprocess.wait()
if p1 and p1.is_alive(): #wait for thread to finish
    p1.join()

已经提出了其他非阻塞读取的解决方案here，但对我不起作用：

1. 需要 readline 的解决方案（包括基于队列的解决方案）总是阻塞。很难（不可能？）杀死执行 readline 的线程。它只会在创建它的进程完成时被杀死，但不会在输出生成进程被杀死时被杀死。
2. 正如 anonnn 指出的那样，将低级 fcntl 与高级 readline 调用混合使用可能无法正常工作。
3. 使用 select.poll() 很简洁，但根据 python 文档，在 Windows 上不起作用。
4. 使用第三方库对于这项任务来说似乎有些矫枉过正，并且会增加额外的依赖项。