使用cStringIO和foreach循环的Python

Question

我想迭代行cStringIO对象，但是它似乎不适用于foreach循环。更准确地说，这种行为就好像集合是空的。我做错了什么？

示例：

Python 2.7.12 (default, Aug 29 2016, 16:51:45)
[GCC 4.4.7 20120313 (Red Hat 4.4.7-3)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import cStringIO
>>> s = cStringIO.StringIO()
>>> import os
>>> s.write("Hello" + os.linesep + "World" + os.linesep)
>>> s.getvalue()
'Hello\nWorld\n'
>>> for line in s :
...     print line
...
>>>

谢谢。

Answer 1

cStringIO.StringIO返回cStringIO.InputType对象(即输入流)(如果提供string else )或返回cStringIO.OutputType对象(即输出流)。

In [13]: sio = cStringIO.StringIO()

In [14]: sio??
Type:        StringO
String form: <cStringIO.StringO object at 0x7f63d418f538>
Docstring:   Simple type for output to strings.

In [15]: isinstance(sio, cStringIO.OutputType)
Out[15]: True

In [16]: sio = cStringIO.StringIO("dsaknml")

In [17]: sio??
Type:        StringI
String form: <cStringIO.StringI object at 0x7f63d4218580>
Docstring:   Simple type for treating strings as input file streams

In [18]: isinstance(sio, cStringIO.InputType)
Out[18]: True

因此，您可以执行读操作或写操作，但不能同时执行这两种操作。对cStringIO.OutputType对象执行读取操作的一个简单解决方案是通过getvalue()方法将其转换为值。

如果尝试同时执行这两个操作，那么它们中的任何一个都会被忽略。

cStringIO.OutputType.getvalue(c_string_io_object)

Answer 2

尝试使用字符串split方法：

for line in s.getvalue().split('\n'): print line
...
Hello
World

或者按照建议，如果您总是在新行上拆分：

for line in s.getvalue().splitlines(): print line

Answer 3

写入后可以从打开的文件句柄读取内容，但首先必须使用seek(0)方法将指针移回起始位置。这将适用于cStringIO或真正的文件：

import cStringIO
s = cStringIO.StringIO()
s.write("Hello\nWorld\n") # Python automatically converts '\n' as needed 
s.getvalue()
# 'Hello\nWorld\n'
s.seek(0)  # move pointer to start of file
for line in s :
    print line.strip()
# Hello
# World