python - 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 尝试以读/写方式模拟一个常规文件，除了它存储在内存中。尝试就地编辑文本文件时遇到与使用常规文件相同的问题(用更长的字符串替换会覆盖以下行)。您也有相同的选择。

其他小变化: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()

它在 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。因此，依赖这样或那样的实现将使您的代码不可移植。