问题

如何像操作文件一样操作str或bytes

类似于C++中的stringstream

解决方案

使用io模块中的StringIO和BytesIO

调用他们的构造函数得到一个io对象

>>> import io
>>> sio = io.StringIO()
>>> bio = io.BytesIO()

然后就可以像文件一样进行read、write、seek等等操作

>>> sio.write('Hello')
5
>>> bio.write(b'World')
5
>>> sio.seek(0)
0
>>> bio.seek(0)
0
>>> sio.read()
'Hello'
>>> bio.read()
b'World'
>>> sio.close()
>>> bio.close()

讨论

当需要模拟一个普通文件而且又不想真正的进行磁盘IO的时候（比如编写单元测试的时候），可以用上这两个类

但是需要注意的是，StringIO和BytesIO并没有真实的文件描述符

并且BytesIO是线程安全的，因为BytesIO内部维护了一个锁来保护公共的资源

而StringIO不是线程安全的

来源

Python CookBook

关注

欢迎关注我的微信公众号：python每日一练