Ajouter une ligne à un fichier existant en Python

Question

J'ai besoin d'ajouter une seule ligne à la première ligne d'un fichier texte et il semble que les seules options disponibles sont plus de lignes de code que ce que j'attendrais de python. Quelque chose comme ceci :

f = open('filename','r')
temp = f.read()
f.close()

f = open('filename', 'w')
f.write("#testfirstline")

f.write(temp)
f.close()

N'y a-t-il pas de solution plus simple ? En outre, je vois plus souvent cet exemple à deux poignées que l'ouverture d'une seule poignée pour la lecture et l'écriture ("r+") - pourquoi ?

Answer 1

Python facilite beaucoup de choses et contient des librairies et des enveloppes pour de nombreuses opérations courantes, mais le but n'est pas de cacher des vérités fondamentales.

La vérité fondamentale que vous rencontrez ici est que vous ne pouvez généralement pas ajouter des données à une structure plate existante sans réécrire l'ensemble de la structure. Cela est vrai quel que soit le langage utilisé.

Il existe des moyens de sauvegarder un filehandle ou de rendre votre code moins lisible, dont beaucoup sont fournis dans d'autres réponses, mais aucun ne change l'opération fondamentale : Vous devez lire le fichier existant, puis écrire les données que vous souhaitez ajouter, suivies des données existantes que vous avez lues.

Vous pouvez vous épargner le filehandle, mais ne cherchez pas à faire tenir cette opération en aussi peu de lignes de code que possible. En fait, ne cherchez jamais à réduire le nombre de lignes de code : c'est de l'obscurcissement, pas de la programmation.

Answer 2

Je m'en tiendrais à des lectures et des écritures séparées, mais nous pouvons certainement les exprimer de manière plus concise :

Python2 :

with file('filename', 'r') as original: data = original.read()
with file('filename', 'w') as modified: modified.write("new first line\n" + data)

Python3 :

with open('filename', 'r') as original: data = original.read()
with open('filename', 'w') as modified: modified.write("new first line\n" + data)

Note : la fonction file() n'est pas disponible dans python3.

Answer 3

Autre approche :

with open("infile") as f1:
    with open("outfile", "w") as f2:
        f2.write("#test firstline")
        for line in f1:
            f2.write(line)

ou un one liner :

open("outfile", "w").write("#test firstline\n" + open("infile").read())

Merci de m'avoir donné l'occasion de réfléchir à ce problème :)

Santé

Answer 4

with open("file", "r+") as f: s = f.read(); f.seek(0); f.write("prepend\n" + s)

Answer 5

Cela permet d'économiser un appel d'écriture :

f.write('#testfirstline\n' + temp)

Lorsque vous utilisez "r+", vous devez rembobiner le fichier après l'avoir lu et avant de l'écrire.