python - Flask View 的流式 shell 输出有效，但永远不会结束

Question

我想将几个 python3 脚本的输出流式传输到我的浏览器。我遵循了来自不同 SO 答案的一些建议并且几乎解决了它。但是在执行脚本后，我从 stdout 读取的 for 循环进入了无限循环。输出是正确的，一切都很好，但是死循环是个问题。如何在输出完成后结束流？

@app.route('/stream/<script>')
def execute(script):
    def inner():
        assert re.match(r'^[a-zA-Z._-]+$', script)
        exec_path = "scripts/" + script + ".py"
        cmd = ["python3", "-u", exec_path]  # -u: don't buffer output

        proc = subprocess.Popen(
            cmd,
            stdout=subprocess.PIPE,
        )

        for line in iter(proc.stdout.readline, ''):
            yield highlight(line, BashLexer(), HtmlFormatter())
            # If process is done, break loop
   #         if proc.poll() == 0:
   #             break

    env = Environment(loader=FileSystemLoader('app/templates'))
    tmpl = env.get_template('stream.html')
    return Response(tmpl.generate(result=inner()))

当我轮询子进程以查看它是否已完成时，脚本的输出被破坏了，因为当打印速度太快时它没有打印整个标准输出流。如果我在每次打印之间添加一个 sleep(1)，问题就不会发生。

代码可以运行，但会进入无限循环。如果我取消注释这些行，输出会以某种方式被缓冲并且部分丢失。我还尝试在一个单独的线程中轮询子进程，并且只需要切换一个标志，如 if not running: break，但这会导致与上面未注释的代码相同的行为。

Answer 1

iter(proc.stdout.readline, '') 调用 readline，直到遇到等于 '' 的内容。但是proc.stdout.readline返回的是bytes对象，而''是一个str对象，所以两者会永远不相等!

相反，要么写

for line in iter(proc.stdout.readline, b''):

或者，甚至更好:

for line in proc.stdout: