監視流類

Question

我想創建一個流對象，在任何時候寫入數據時觸發一個回調函數。

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，完全繞過寫入功能。有沒有辦法可以監控這個輸出呢？

Answer 1

我相信這裏的問題是，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()) 

雖然我覺得這是不太優雅。