问题
如何像操作文件一样操作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每日一练