在Django StreamHttpResponse中实时逐行打印子进程stdout

Question

我在 Django 中写了一个 API。 它在服务器中运行 shell 脚本并打印一些日志。 当客户端在浏览器中调用API时，我希望浏览器逐行打印日志。 到目前为止，代码片段如下：

from django.http import StreamingHttpResponse
import subprocess
import shlex


def test(request):
    cmd = 'bash test.sh'
    args = shlex.split(cmd)
    proc = subprocess.Popen(
        args,
        cwd='/foo/bar',
        shell=False,
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
    )
    return StreamingHttpResponse(
        (line for line in iter(proc.stdout.readline, '')),
        content_type="text/plain",
    )

而test.sh就像

#!/bin/bash

echo hello
sleep 1
echo world
sleep 1
echo love
sleep 1
echo and
sleep 1
echo peace
sleep 1

在bash test.sh完成之前不会打印日志。 我怎样才能让它打印日志，就好像test.sh在客户端运行一样？

我正在使用 Python 2.7 和 Django 1.11.29。 我知道它们很旧，但我现在仍然必须使用它们。 感谢帮助。

Answer 1

您将内容提供给StreamingHttpResponse为：

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

在这里，您创建一个可迭代的并同时迭代它并返回一个生成器。 虽然这并不重要，你应该得到实时 output。 问题可能只是 output 太快而无法实际看到实时差异，或者您的配置可能不正确（请参阅流式传输 HTTP 响应，刷新到浏览器），另一个问题可能是您没有将响应用作stream 和 javascript 而是在使用前完全下载它。 您可以尝试使用time.sleep()来查看它是否太快或配置问题：

import time


def iter_subproc_with_sleep(proc):
    for line in iter(proc.stdout.readline, ''):
        time.sleep(1)
        yield line

def test(request):
    cmd = 'bash test.sh'
    args = shlex.split(cmd)
    proc = subprocess.Popen(
        args,
        cwd='/foo/bar',
        shell=False,
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
    )
    return StreamingHttpResponse(
        iter_subproc_with_sleep(proc),
        content_type="text/plain",
    )