2

我正在尝试创建一个流对象，该对象在数据写入任何时候触发回调函数。

class MonitoredStream():
    def __init__(self, outstream, callback):
        self.outstream = outstream
        self.callback = callback

    def write(self, s):
        self.callback(s)
        self.outstream.write(s)

    def __getattr__(self, attr):
        return getattr(self.outstream, attr)

当我直接调用 write 方法时，这很好用，但是当我有一个子进程的输出连接到流时，我希望它也能正常工作。例如：

def f(s):
    print("Write")

p = sub.Popen(["sh", "test.sh"], stdout=MonitoredStream(sys.stdout, f))
p.communicate()

这只是将输出直接发送到 sys.stdout，完全绕过了 write 函数。有没有办法我也可以监控这个输出？

4

1 回答 1

2

我相信这里的问题是它subprocess.Popen不使用管道的 Python 接口 - 它而是获取文件描述符，然后使用它直接写入管道，当你给出stdout管道的属性时，这意味着它使用它，绕过你的代码。

解决这个问题我最好的猜测是在中间制作一个新的中间管道，让您自己处理流。我会将其实现为上下文管理器：

import sys
import os
from subprocess import Popen
from contextlib import contextmanager

@contextmanager
def monitor(stream, callback):
    read, write = os.pipe()
    yield write
    os.close(write)
    with os.fdopen(read) as f:
        for line in f:
            callback(line)
            stream.write(line)

def f(s):
    print("Write")

with monitor(sys.stdout, f) as stream:
    p = Popen(["ls"], stdout=stream)
    p.communicate()

当然，尽管您仍然可以使用类：

import sys
import os
from subprocess import Popen

class MonitoredStream():
    def __init__(self, stream, callback):
        self.stream = stream
        self.callback = callback
        self._read, self._write = os.pipe()

    def fileno(self):
        return self._write

    def process(self):
        os.close(self._write)
        with os.fdopen(self._read) as f:
            for line in f:
                self.callback(line)
                self.stream.write(line)

def f(s):
    print("Write")

stream = MonitoredStream(sys.stdout, f)
p = Popen(["ls"], stdout=stream)
p.communicate()
print(stream.process())

虽然我觉得这不太优雅。

于 2012-06-08T19:15:59.123 回答