标准流重定向为Python对象
前面介绍过Python流的重定向,但都是在命令行窗口中实现的,不是在同一个Python脚本中做到的。如何在同一个Python脚本中实现标准流的重定向呢?这是本文将要探讨的问题。
标准流对象是在运行Python脚本或者启动Python控制台窗口时通过open函数创建的文件对象的引用。既然是引用,就可以重新进行赋值。被赋值的对象需要满足以下要求:
- 任意提供read()方法的对象都可以赋值给sys.stdin,这样调用sys.stdin.read()就会调用该对象的read()获取数据
- 任意提供write()方法的对象都可以赋值给sys.stdout,这样所有的标准输出都可以发送到该对象的方法中。
这样的对象又称为仿文件对象(file-like object)。
在Python的官方文档中有这样一句描述:“If it looks like a duck and quacks like a duck, it must be a duck.”(看起来像一只鸭子,叫声也像一只鸭子,那么它就是一只鸭子——有点像绕口令。)这种性质在Python中称为
polymorphism。从编程的角度来解释,就是如果一类对象它们的属性名和属性行为相似,那么就可以把它们当做相同的类型来使用,无论这些对象是函数,还是类。
根据以上原则,仿文件对象就是文件对象,可以通过read()/readline()/write()等文件方法进行文本的读写操作。
需要指明的是:
- stdin处理所有用户交互的输入(包括调用内建函数input(),同样也会使用stdin)
- stdout处理print()函数的输出(控制台的回显估计也是用这个函数实现的)和input()表达式中的输出(input(str),str不为None时,函数内部会调用print(str)打印提示信息,再等待用户的输入,用户输入完之后还会回显)
- stderr处理Python解析器自身的指令和错误信息
- 在版本2.7.2中,input(),raw_input()函数内部会调用sys.stdin.readline()方法;在版本3.x中,raw_input()和input()的功能合并,只保留了input(),input()函数内部仍然会调用sys.stdin.readline()方法。而print()函数内部调用sys.stdout.write()方法。
可以通过程序进行简单的测试(测试版本Python 3.3.5 shell)。
>>> class Input(object):
def __init__(self, inStr = ''):
self.txt = inStr
def read(self, size = None):
if size:
block, self.txt = self.txt[:size], self.txt[size:]
else:
block, self.txt = self.txt, ''
return block
>>> sys.stdin = Input('hello,world')
>>> input('Enter:')
Enter:Traceback (most recent call last):
File "<pyshell#6>", line 1, in <module>
input('Enter:')
AttributeError: 'Input' object has no attribute 'readline'
由于类Input没有提供readline方法,所以执行input()时失败
>>> class Input(object):
def __init__(self, inStr = ''):
self.txt = inStr
def read(self, size = None):
if size:
block, self.txt = self.txt[:size], self.txt[size:]
else:
block, self.txt = self.txt, ''
return block
def readline(self):
eoln = self.txt.find('\n')
if eoln == -1:
line, self.txt = self.txt, ''
else:
line, self.txt = self.txt[:eoln + 1], self.txt[eoln + 1:]
return line
>>> sys.stdin = Input('hello,world')
>>> input("Enter: ")
Enter: 'hello,world'
以上脚本在v2.7.2的控制台窗口中执行时,必须用raw_input()代替input(),不然会报错。
将类的对象赋值给sys.stdin,使得input()直接调用类的readline()方法。不过如果只调用input(),则类中含有readline()方法就可以了,如:
>>> class Input:
def readline(self):
return "hello,world"
>>> sys.stdin = Input()
>>> input("Enter: ")
Enter: 'hello,world'
需要注意的是在控制台修改sys.stdin/sys.stdout引用之前,最好先保存下原有的流对象,对流对象操作结束之后,再将保存的对象替换回去。如果不这样,就会导致控制台无法回显;quit(),exit()等方法也会失效。如下面的例子:
>>> import sys
>>> save = sys.stdout
>>> class Output(object):
def __init__(self):
self.txt = ''
def write(self, txt):
self.txt += txt
def writelines(self, txt):
string = ''.join(txt)
self.write(txt)
>>> obj = sys.stdout = Output()
>>> print("hello,world")
>>> 1 + 2
>>> "ni hao"
>>> sys.stdout = save
>>> obj.txt
"hello,world\n3\n'ni hao'\n"
可以看出修改了sys.stdout之后,控制台无法回显,需要回显的字符都存储在了obj.txt中。只有在恢复stdout的本来流对象之后,才可以正常回显了。
Python中的io模块提供了工具类StringIO和BytesIO,自动完成了上述的重定向的工作。可以不用自己敲代码了,因而十分便利。
例如:
例如:
>>> import sys
>>> temp = sys.stdout
>>> buffer = StringIO()
>>> sys.stdout = buffer
>>> print("hello,","world")
>>> sys.stdout = temp
>>> buffer.getvalue()
'hello, world\n'
Print()函数的重定向
print()函数在默认的情况下将文本输出到sys.stdout中,可以用一个仿文件对象替换它进行重定位。这样不用修改sys.stdout也可达到前文提到的重定位的效果。具体演示如下:>>> class Output:
def __init__(self):
self.txt = ''
def write(self, Str):
self.txt += Str
def getValue(self):
return self.txt
>>> buffer = Output()
>>> print("hello", file = buffer)
>>> print("world", file = buffer)
>>> buffer.getValue()
'hello\nworld\n'
os.popen()的重定向
第三节中提到popen()方法创建对象时的模式,默认情况下是只读的("r"),可以修改模式为"w"。这样,可以在Python脚本中或者控制台窗口给命令行中的脚本传递文本信息,而无需借助命令行参数。
test_popen.py:
"""test the os.popen"""
def Test():
open(r'D:\1.txt').write((input()) #不要使用print(input())或者sys.stdout.write(input())
Test()
在MS-DOS窗口执行Python,调出Python命令行窗口,然后敲入以下代码:
>>> import os
>>> f = os.popen('python test_popen.py', 'w')
>>> f.write('hello,world')#将"hello,world"传递给脚本test_popen.py,因为f.write会修改sys.stdout的行为,所以会导致print()和sys.stdout.write()函数失效
11
>>> f.close()
>>>
打印文件"D:\1.txt"的内容:
os.popen()的重定向和前面提到的标准流的重定向意义是不一样的。标准流的重定向是将输入输出流的目标指定为某个Python对象,这种重定向发生在同一个进程内;os.popen()可以实现不同进程间的信息通信。比如上例中,通过设定写模式,可以将Python控制台的文本传递test_popen.py脚本,运行时,它们分属两个不同进程。类似的Python函数还有subprocess.call()和subprocess.Popen(),比起os.popen(),函数subprocess.Popen在进程通信的控制上更为细致。
可以通过控制台传递文本给脚本:
也可以将脚本执行的结果传递给控制台:
print_helloworld.py:
print "hello,world"
控制台的执行结果:
脚本之间相互传递文本: