如何拦截Python中stdout的内容？

Question

我在Django中做了一个自定义的命令，在那里向stdout进行一些输出。 如何拦截stdout进行一些更改（例如添加前缀）并写入stderr？ 这是测试任务，所以不要关注问题的不合逻辑性。

Answer 1

您可以使用StringIO这样的：

from StringIO import StringIO 
import sys 

old_stdout = sys.stdout 
sys.stdout = mystdout = StringIO() 

print("test") 

sys.stdout = old_stdout 

s = mystdout.getvalue() 
s = "prefix" + s 

sys.stderr.write(s) 

的代码捕获stdout到缓冲区，然后读取缓冲区，增加prefix并写入stderr。

另见：

• Can I redirect the stdout in python into some sort of string buffer?

Answer 2

您可以使用sys.stdout和sys.stderr：

from sys import stdout, stderr 
output_text = "Some text" 
stdout.write('stdout: ' + output_text + '\n') 
stderr.write('stderr: ' + output_text + '\n') 

演示：

monty@xonix:~/py$ python so.py &> foo.txt 
monty@xonix:~/py$ cat foo.txt 
stderr: Some text 
stdout: Some text