python - 搜索文本文件并插入行

Question

我要做的是（以下面的文本为例），在文本文件中搜索字符串“Text2”，然后在“Text 2”后两行插入一行（“Inserted Text”）。“文本 2”可以在文本文件中的任何一行，但我知道它会在文本文件中出现一次。

所以这是原始文件：

Text1
Text2
Text3
Text4

这就是我想要的：

Text1
Text2
Text3
Inserted Text
Text 4

所以我已经知道如何使用下面的代码在一行上方添加文本。

for line in fileinput.input('file.txt', inplace=1,backup='.bak'):
    if line.startswith('Text 4'):
        print "Inserted Text"
        print line,
    else:
        print line,

但我只是不知道如何在文件中搜索的文本之后添加两行内容。

Answer 1

快速肮脏的方式就是这样

before=-1
for line in fileinput.input('file.txt', inplace=1,backup='.bak'):
    if line.startswith('Text 2'):
        before = 2
    if before == 0
        print "Inserted Text"
    if before > -1
        before = before - 1
    print line,

Answer 2

如果将文件内容加载到列表中，操作起来会更容易：

searchline = 'Text4'
lines = f.readlines() # f being the file handle
i = lines.index(searchline) # Make sure searchline is actually in the file

现在i包含行的索引Text4。您可以使用它并list.insert(i,x)在之前插入：

lines.insert(i, 'Random text to insert')

或之后：

lines.insert(i+1, 'Different random text')

或后三行：

lines.insert(i+3, 'Last example text')

只要确保包含IndexErrors 的错误处理，您就可以随心所欲地处理它。

Answer 3

你可以使用

f = open("file.txt","rw")
lines = f.readlines()
for i in range(len(lines)):
     if lines[i].startswith("Text2"):
            lines.insert(i+3,"Inserted text") #Before the line three lines after this, i.e. 2 lines later.

print "\n".join(lines)