Question

这是我的问题。我有一个应用程序，它使用logging模块将一些迹线打印到标准输出。现在，我希望能够同时读取这些跟踪，以便等待我需要的特定跟踪。

这是出于测试目的。因此，例如，如果在大约2秒钟内没有发生所需的跟踪，则测试将失败。

我知道我可以使用类似这样的内容来读取其他脚本的输出：

import subprocess
p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
while True:
    line = p.stdout.readline()
    print line
    if line == '' and p.poll() != None:
        break

但是，如何从脚本本身做类似的事情？预先感谢。

编辑

因此，由于我的问题是希望在运行Python应用程序时出现某些跟踪记录，并且由于无法从应用程序本身中找到一种简单的方法，因此我决定启动该应用程序（如注释中所建议））中的其他脚本。

我发现非常有用的模块是subprocess模块，它比pexpect模块更易于使用。

Answer 1

如果要对记录器消息进行一些预处理，可以执行以下操作：

#!/usr/bin/python

import sys
import logging
import time
import types

def debug_wrapper(self,msg):
  if( hasattr(self,'last_time_seen') and 'message' in msg):
    print("INFO: seconds past since last time seen "+str(time.time()-self.last_time_seen))
  self.last_time_seen = time.time()
  self.debug_original(msg)

logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
logger = logging.getLogger("test")

logger.debug_original = logger.debug
logger.debug = types.MethodType(debug_wrapper, logger)

while True:
  logger.debug("INFO: some message.")
  time.sleep(1)

这可以通过使用自定义的debug_wrapper函数替换记录器对象的原始调试函数来实现，在该函数中，您可以执行所需的任何处理，例如，存储上次看到消息的时间。

Answer 2

您可以将脚本输出实时存储到文件中，然后实时读取脚本中的内容（因为输出文件中的内容会动态更新）。

要将脚本输出实时存储到文件中，可以使用Expect包随附的unbuffer。

sudo apt-get install expect

然后，在运行脚本时使用：

unbuffer python script.py > output.txt

您只需要在脚本中打印输出，该脚本将动态更新为输出文件。因此，每次都读取该文件。

此外，使用>覆盖旧文件或创建新文件，并使用>>将内容附加到先前创建的output.txt文件中。

Answer 3

如果要用其他Python代码记录print语句的输出，可以将sys.stdout重定向到类似于文件对象的字符串，如下所示：

import io
import sys

def foo():
    print("hello world, what else ?")

stream = io.StringIO()
sys.stdout = stream
try:
    foo()
finally:
    sys.stdout = sys.__stdout__

print(stream.getvalue())

如何从其中读取Python脚本的标准输出？

3 个答案: