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 模块的 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() 的非阻塞替代方案，并且基于此答案：

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()

此处提出了其他非阻塞读取解决方案，但对我不起作用：

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