python stdout 重定向

最新推荐文章于 2024-06-02 16:09:13 发布

weixin_33978016

最新推荐文章于 2024-06-02 16:09:13 发布

阅读量116

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/hangj/p/4990719.html

本文介绍了一个使用Python自定义类来捕获和重定向标准输出到内存缓冲区的方法。通过这个技巧，可以方便地捕获程序运行时的标准输出，并进一步处理这些输出信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

 1 import sys
 2 
 3 class Buffer(object):
 4     def __init__(self):
 5         self.buffer = []
 6 
 7     def write(self, *args, **kwargs):
 8         self.buffer.append(args)
 9 
10 def test():
11     stdout = sys.stdout
12     sys.stdout = open('hello_stdout', 'wb')
13 
14     print 'test'
15 
16     sys.stdout.close()
17     sys.stdout = stdout
18 
19 
20     stdout = sys.stdout
21     sys.stdout = Buffer()
22 
23     print 'buffer1'
24     print 'buffer2'
25 
26     buff, sys.stdout = sys.stdout, stdout
27     print buff.buffer
28 
29 if __name__ == '__main__':
30     test()