python - 如何从Python脚本中读取标准输出？

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 应用程序运行时出现某些跟踪，并且由于我无法从应用程序本身找到一种简单的方法来执行此操作，因此我决定启动该应用程序(如注释中所建议的那样) )来自另一个脚本。

我发现 pexpect 模块非常有用，并且比 subprocess 模块更容易使用。

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 函数替换记录器对象的原始调试函数来实现的，您可以在其中执行您想要的任何处理，例如存储您上次看到消息的时间。