Rechercher et remplacer une ligne dans un fichier en Python

Question

Je veux boucler sur le contenu d'un fichier texte, effectuer une recherche et un remplacement sur certaines lignes et réécrire le résultat dans le fichier. Je pourrais d'abord charger tout le fichier en mémoire, puis le réécrire, mais ce n'est probablement pas la meilleure façon de procéder.

Quelle est la meilleure façon de procéder, dans le code suivant ?

f = open(file)
for line in f:
    if line.contains('foo'):
        newline = line.replace('foo', 'bar')
        # how to write this newline back to the file

Answer 1

Le moyen le plus court serait probablement d'utiliser la fonction module d'entrée de fichiers . Par exemple, l'exemple suivant ajoute des numéros de ligne à un fichier, en place :

import fileinput

for line in fileinput.input("test.txt", inplace=True):
    print "%d: %s" % (fileinput.filelineno(), line),

Ce qui se passe ici est :

1. Le fichier original est déplacé vers un fichier de sauvegarde
2. La sortie standard est redirigée vers le fichier original à l'intérieur de la boucle.
3. Ainsi, tout print les instructions réécrivent dans le fichier original

fileinput a plus de cloches et de sifflets. Par exemple, il peut être utilisé pour opérer automatiquement sur tous les fichiers de la section sys.args[1:] sans que vous ayez à les itérer explicitement. À partir de Python 3.2, il fournit également un gestionnaire de contexte pratique à utiliser dans un environnement with déclaration.

---

Alors que fileinput est idéal pour les scripts à jeter, mais je me méfierais de son utilisation dans du code réel, car il est vrai qu'il n'est pas très lisible ou familier. Dans du code réel (de production), il vaut la peine de passer quelques lignes de code supplémentaires pour rendre le processus explicite et donc rendre le code lisible.

Il existe deux options :

1. Le fichier n'est pas trop volumineux, et vous pouvez le lire entièrement en mémoire. Fermez ensuite le fichier, rouvrez-le en mode écriture et réécrivez le contenu modifié.
2. Le fichier est trop volumineux pour être stocké en mémoire ; vous pouvez le déplacer vers un fichier temporaire et l'ouvrir, le lire ligne par ligne et le réécrire dans le fichier d'origine. Notez que cela nécessite deux fois plus de stockage.

Answer 2

Je suppose que quelque chose comme ça devrait le faire. En gros, il écrit le contenu dans un nouveau fichier et remplace l'ancien fichier par le nouveau :

from tempfile import mkstemp
from shutil import move
from os import remove, close

def replace(file_path, pattern, subst):
    #Create temp file
    fh, abs_path = mkstemp()
    new_file = open(abs_path,'w')
    old_file = open(file_path)
    for line in old_file:
        new_file.write(line.replace(pattern, subst))
    #close temp file
    new_file.close()
    close(fh)
    old_file.close()
    #Remove original file
    remove(file_path)
    #Move new file
    move(abs_path, file_path)

Answer 3

Voici un autre exemple qui a été testé et qui correspond aux modèles de recherche et de remplacement :

import fileinput
import sys

def replaceAll(file,searchExp,replaceExp):
    for line in fileinput.input(file, inplace=1):
        if searchExp in line:
            line = line.replace(searchExp,replaceExp)
        sys.stdout.write(line)

Exemple d'utilisation :

replaceAll("/fooBar.txt","Hello\sWorld!$","Goodbye\sWorld.")

Answer 4

Cela devrait fonctionner : (édition sur place)

import fileinput

for line in fileinput.input(files, inplace = 1): # Does a list of files, and writes redirects STDOUT to the file in question
      print line.replace("foo", "bar"),

Answer 5

Basé sur la réponse de Thomas Watnedal. Cependant, cela ne répond pas exactement à la partie ligne à ligne de la question initiale. La fonction peut toujours remplacer, ligne par ligne, les éléments suivants

Cette mise en œuvre remplace le contenu du fichier sans utiliser de fichiers temporaires, par conséquent, les autorisations de fichiers restent inchangées.

De même, re.sub au lieu de replace, permet le remplacement par regex au lieu du remplacement par texte brut uniquement.

La lecture du fichier en tant que chaîne unique plutôt que ligne par ligne permet une correspondance et un remplacement multilignes.

import re

def replace(file, pattern, subst):
    # Read contents from file as a single string
    file_handle = open(file, 'r')
    file_string = file_handle.read()
    file_handle.close()

    # Use RE package to allow for replacement (also allowing for (multiline) REGEX)
    file_string = (re.sub(pattern, subst, file_string))

    # Write contents to file.
    # Using mode 'w' truncates the file.
    file_handle = open(file, 'w')
    file_handle.write(file_string)
    file_handle.close()