StringIO学习

StringIO

StringIO的行为与file对象非常像，但它不是磁盘上文件，而是一个内存里的“文件”，我们可以将操作磁盘文件那样来操作StringIO。一个简单的例子，让你对StringIO有一个感性的认识：

from StringIO import StringIO

# 生成一个StringIO对象，当前缓冲区内容为ABCDEF  
s = StringIO('ABCDEF')
# 从开头写入，将会覆盖ABC
s.write('abc')
# 每次使用read()读取前，必须seek()
# 定位到开头
s.seek(0)
# 将输出abcDEF
print s.read()
# 定位到第二个字符c
s.seek(2)
# 从当前位置一直读取到结束，将输出cDEF
print s.read()
s.seek(3)
# 从第三个位置读取两个字符，将输出DE
print s.read(2)
s.seek(6)
# 从指定位置写入
s.write('GH')
s.seek(0)
# 将输出abcDEFGH
print s.read()
# 如果读取所有内容，可以直接使用getvalue()
# 将输出abcDEFGH
print s.getvalue()

通过例子，我们看到了StringIO的行为，基本与file一致。StringIO提供了一个方法，可以方便的获取其中的数据：StringIO. getvalue()。如果使用read方法获取其中的数据，必须通过seek先设置"文件指针"的位置。

 

cStringIO

Python标准模块中还提供了一个cStringIO模块，它的行为与StringIO基本一致，但运行效率方面比StringIO更好。因此，import的时候会写成下面的方式，但使用 cStringIO模块时，有几个注意点： 

1. cStringIO.StringIO不能作为基类被继承；

2. 创建cStringIO.StringIO对象时，如果初始化函数提供了初始化数据，新生成的对象是只读的。

try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO

s = StringIO('A')
s.write('abc')
# 会提示：AttributeError: 'cStringIO.StringI' object has no attribute 'write'
print s.getvalue()