Python zu viel schreiben

Question

Ich habe mich im Internet umgesehen und kann keine Antwort finden, tut mir leid, wenn ich es verpasst habe.

Ich versuche, einfach einige Zeilen in einer Datei zu ersetzen, etwas mit dieser modifizierten Datei zu tun, dann schreiben Sie es in den ursprünglichen Zustand. Also behalte ich eine Kopie der Basisdatei, ändere die, in der ich etwas tun muss, schreibe die Basisdatei in die modifizierte um.

Hier ist mein Code:

lines = open('fileToModify').readlines() 
for i, line in enumerate(lines[:]): 
    if "STRING" in line: 
     lines[i] = line.replace("STRING", sys.argv[1]) 
    if "STRING2" in line: 
     lines[i] = line.replace("STRING2", sys.argv[2]) 
pfile = open("fileToModify", "r+") 
pfile.seek(0) 
pfile.writelines(lines) 
pfile.seek(0) 
baseLines = open("baseFile").readlines() 
pfile.writelines(baseLines) 
pfile.close() 

Das Problem ist, dass, wenn ich die Basisdatei zu dem modifizierten eine neu zu schreiben, die letzten 2 Zeichen der Basisdatei in eine neue Zeile der modifizierten Datei kopiert werden. Ich weiß nicht, was ich falsch mache, es kommt mir wirklich komisch vor ...

Auch wenn es einen besseren Weg gibt, diesen Zweck zu erreichen, bin ich bei allen Vorschlägen dabei.

Vielen Dank im Voraus!

Answer 1

from shutil import copy 

s1, r1 = "STRING", sys.argv[1] 
s2, r2 = "STRING2", sys.argv[2] 

with open("orig_file") as in_f, open("new_file", "w") as out_f: 
    for line in in_f: 
     out_f.write(line.replace(s1, r1).replace(s2, r2)) 

copy("new_file", "orig_file")