Python: Öffnen Sie eine Datei, suchen Sie dann anhängen, wenn nicht existieren

Question

Ich versuche, eine Zeichenfolge an eine Datei anhängen, wenn die Zeichenfolge in der Datei nicht beendet wird. Das Öffnen einer Datei mit der Option a+ ermöglicht es mir jedoch nicht, dies sofort zu tun, da das Öffnen der Datei mit a+ den Zeiger an das Ende der Datei setzt, was bedeutet, dass meine Suche immer fehlschlägt. Gibt es einen guten Weg, dies zu tun, außer die Datei zu öffnen, um zuerst zu lesen, zu schließen und wieder zum Anhängen zu öffnen?

In Code, scheinbar, unten funktioniert nicht.

file = open("fileName", "a+") 

Ich muss folgendes tun, um es zu erreichen.

file = open("fileName", "r") 
... check if a string exist in the file 
file.close() 

... if the string doesn't exist in the file 
file = open("fileName", "a") 
file.write("a string") 
file.close() 

Answer 1

die Eingabedatei unverändert zu lassen, wenn needle auf jeder Linie ist oder die Nadel am Ende der Datei anhängen, wenn sie fehlt:

with open("filename", "r+") as file: 
    for line in file: 
     if needle in line: 
      break 
    else: # not found, we are at the eof 
     file.write(needle) # append missing data 

Ich habe es getestet und es funktioniert sowohl auf Python 2 (stdio-basierte I/O) und Python 3 (POSIX lesen/schreiben-basierte I/O).

Der Code verwendet obskuren else nach einer Schleife Python-Syntax. Siehe Why does python use 'else' after for and while loops?

Answer 2

Sie können die aktuelle Position des Dateiobjekts mithilfe von file.seek() festlegen. Zu springen, um den Anfang einer Datei verwenden

f.seek(0, os.SEEK_SET) 

zu einer Datei Ende zu springen, verwenden

f.seek(0, os.SEEK_END) 

In Ihrem Fall, um zu überprüfen, ob eine Datei etwas enthält, und dann hängen Sie vielleicht append die Datei, würde ich so etwas tun:

import os 

with open("file.txt", "r+") as f: 
    line_found = any("foo" in line for line in f) 
    if not line_found: 
     f.seek(0, os.SEEK_END) 
     f.write("yay, a new line!\n")