更改文件中的整行

Question

我的程序必须通过第一个参数（312,621）在文件中取一行并更改第一行或两个参数。之后必须再次写入文件。 第一个参数是唯一且不可更改的

之前的文件：

312|fotelja snesko|white
621|digimon tabure|black

我的代码：

key = input("key >>")

entitet = open("komad_namestaja.txt", "r+")
allDATA = entitet.readlines()
entitet.close()
for line in allDATA:
    lineLST = line.split("|")

    while key == lineLST[0]:

        print("name", lineLST[1])
        name = input("name >> ")
        if name == "":
            name = lineLST[1]

        print("color", lineLST[2])
        color = input("color >> ")
        if color == "":
            color = lineLST[2]

键输入：312

姓名输入：cokoloko

颜色输入：蓝色

文件之后：

312|cokoloko|blue
621|digimon tabure|black

问题： 如何使用之前的行重写新输入，而不会丢失第二行。

Answer 1

尝试查看fileinput模块

for line in fileinput.input("input_file.txt"):
    if check(line):
       print line.replace("red","blue")
    else:
       print line