Script Execution Context(Chapter 3 of Programming Python)

最新推荐文章于 2025-09-12 15:58:40 发布

weixin_30823001

最新推荐文章于 2025-09-12 15:58:40 发布

阅读量88

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/zhtf2014/archive/2011/04/05/2005593.html

本文介绍了一种在Python中重定向标准输入输出的方法，通过自定义Input和Output类实现，能够在函数调用期间捕获输入输出，适用于测试或日志记录等场景。

1 import sys
2
3 class Output:
4     def __init__(self):
5         self.text = ''
6     def write(self, string):
7         self.text += string
8     def writelines(self, lines):
9         for line in lines:
10             self.write(line)
11
12 class Input:
13     def __init__(self, input=''):
14         self.text = input
15     def read(self, size=None):
16         if size == None:
17             res, self.text = self.text, ''
18         else:
19             res, self.text = self.text[:size], self.text[size:]
20     def readline(self):
21         eoln = self.text.find('\n')
22         if eoln == -1:
23             res. self.text = self.text, ''
24         else:
25             res, self.text = self.text[:eoln+1], self.text[eoln+1:]
26
27 def redirect(function, pargs, kargs, input):
28     savestreams = sys.stdin, sys.stdout
29     sys.stdin = Input(input)
30     sys.stdout = Output()
31     try:
32         result = function(*pargs, **kargs)
33     finally:
34         sys.stdin, sys.stdout = savestreams
35     return (result, output)

转载于:https://www.cnblogs.com/zhtf2014/archive/2011/04/05/2005593.html