python 内存中的文件操作 StringIO cStringIO 简介

系统管理员 2021-09-01 07:23 242阅读 0赞

### 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()