python - 在特定行之后添加行

Question

我正在尝试将特定行添加到文件中的特定区域。我正在使用这个:

new_file = open("file.txt", "r+")
 for line in new_file:
  if line == "; Include below":
     line = line + "\nIncluded text"
     new_file.write(line)
  else:
     new_file.write(line)

但出于某种原因，我的 file.txt 的内容是重复的。

编辑:如果我的文件看起来像:

blablablablablablabal
balablablabalablablbla
include below
blablablablablabalablab
ablablablabalbalablaba

我想让它看起来像:

blablablablablablabal
balablablabalablablbla
include below
included text
blablablablablabalablab
ablablablabalbalablaba

Answer 1

在读取文件时不能安全地写入文件，最好将文件读入内存，更新它，然后将其重写到文件。

with open("file.txt", "r") as in_file:
    buf = in_file.readlines()

with open("file.txt", "w") as out_file:
    for line in buf:
        if line == "; Include this text\n":
            line = line + "Include below\n"
        out_file.write(line)