python - 读取文件并写入我刚刚读取的行

Question

所以我想通读一个文件，检查每一行，看看它是否包含我正在查找的字符串的一部分。一旦找到正确的行，我就想在文件中重写该行。

这是我到目前为止所拥有的:

f = open("playlist.py", "r+")
for line in f:
    if old in line:
        f.write("    " + str(item) + ":" + " " + 
                "\""  + new_text + "\"") 
        f.close()
        break
f.close()

此代码正在查找正确的行，但写入文件末尾。我有点认为读写迭代器会被共享，但我想不是:(

Answer 1

使用文件输入

import fileinput

for line in fileinput.input("test.txt", inplace=True):
    if contains_str_youlookingfor:  
        print "what_you_want_to_rewire"
    else:
        print line.rstrip()
fileinput.close()

如果您使用的是 Python 3.2+，则首选上下文管理器:

with fileinput.input("test.txt", inplace=True) as f:    
    for line in f:
        if contains_str_youlookingfor:  
            print("what_you_want_to_rewire")
        else:
            print(line, end='')

这些行将被替换为“what_you_want_to_rewire”，并且不会创建新文件。

编辑
如果您不删除原始换行符或在没有换行符的情况下进行打印，那么您将得到额外的空行。