StringIO 中的就地替换

Question

如何用 StringIO 中的另一个字符串替换另一个字符串？ - 我听说如果它们的长度相同是可能的。

尝试：

from cStringIO import StringIO

c = 'can\nhaz\nfoo'

sio = StringIO(c)

for line in sio:
    if line == 'haz\n':
        # sio.write('bar\n')
        line = 'bar\n'
        break

sio.seek(0)
sio.readlines()   # [ 'can\n', 'haz\n', 'bar' ]

PS：目前正在研究 C 中的解决方案，但更愿意让它工作。

Answer 1

StringIO 尝试模拟 read/write 中的常规文件，只是它存储在内存中。尝试就地编辑文本文件时遇到与使用常规文件相同的问题（用更长的字符串替换会覆盖以下行）。您也有相同的选择。

其他小改动：cStringIO 在开始时使用默认字符串创建是不可写的。但是你可以通过创建一个空对象并写入它来解决这个问题（你现在显然可以访问 write 方法）

您要查找的内容：

c = 'can\nhaz\nfoo'

sio = StringIO()
sio.write(c)
sio.seek(0)

offset = sio.tell()
for line in sio:
    if line == 'haz\n':
        sio.seek(offset)
        sio.write('bar\n')
        break
    offset = sio.tell()

它存储偏移量 before lineread，如果找到模式，它会返回到以前的偏移量，并写入相同数量的字节

如果你想使用 cStringIO 的残酷解决方案：你可以完全读取缓冲区，执行替换，然后写回它（允许用更长的字符串替换），但读取整个缓冲区，这是可能不是你想要的：

c = 'can\nhaz\nfoo'

sio = StringIO(c)

v = sio.getvalue()
v = v.replace("haz","bar")
sio.seek(0)
sio.write(v)

请注意，Python 3 删除了那些对象，现在只是 io。因此，依赖这样或那样的实现将使您的代码不可移植。

StringIO 中的就地替换

in-place replacement in StringIO

python

python-2.7

python-internals

stringio

cstringio