Abrir archivo, leerlo, procesarlo y escribir de nuevo - el método más corto en Python

Question

Quiero hacer algunos filtros básicos en un archivo. Léelo, hazlo, escríbelo.

No estoy buscando "jugar al golf", pero quiero el método más simple y elegante para lograrlo. Se me ocurrió:

from __future__ import with_statement 

filename = "..." # or sys.argv... 

with open(filename) as f: 
    new_txt = # ...some translation of f.read() 

open(filename, 'w').write(new_txt) 

La declaración with hace las cosas más corto ya que no tiene que abrir de forma explícita y cierre el archivo.

¿Alguna otra idea?

Answer 1

realidad una forma más fácil usando fileinput es utilizar el parámetro in-situ:

import fileinput 
for line in fileinput.input (filenameToProcess, inplace=1): 
    process (line) 

Si usa el parámetro inplace, redireccionará stdout a su archivo, de modo que si imprime, lo volverá a escribir en su archivo.

este ejemplo se agrega números de línea a su archivo:

import fileinput 

for line in fileinput.input ("b.txt",inplace=1): 
    print "%d: %s" % (fileinput.lineno(),line), 

Answer 2

Esto parece funcionar:

with open(filename, "r+") as f: 
    new_txt = process(f.read()) 
    f.truncate(0) 
    f.write(new_txt) 

Answer 3

me gustaría ir por la elegancia de una manera diferente: poner en práctica sus operaciones de archivo de lectura y filtrado como generadores, usted va a escribir más líneas de código, pero será más código flexible, mantenible y de rendimiento.

Consulte David M. Beazley's Generator Tricks for Systems Programmers, que es algo muy importante para cualquier persona que escriba este tipo de código para leer.

Answer 4

Si usted está buscando el pitón equivalente de "perl-pi", aquí es una muy buena:

 
import fileinput 
for line in fileinput.input(): 
    # process line 

Ver http://www.python.org/doc/2.5.2/lib/module-fileinput.html por más.

Hecho de esta manera, se utiliza la secuencia de comandos de Python en un tubo para crear el nuevo archivo:

 
$ myscript.py infile.txt > outfile.txt 

Answer 5

para hacerlo de una manera que no se eat your data si te estrellas en el centro:

from twisted.python.filepath import FilePath 
p = FilePath(filename) 
p.setContent(process(p.getContent())) 

Answer 6

Mi fea (pero corto como se indica en la pregunta) solución con generator expressions;

# Some setup first 
file('test.txt', 'w').write('\n'.join('%05d' % i for i in range(100))) 


# This is the filter function 
def f(i): 
    return i % 3 


# This is the main part 
file('test2.txt', 'w').write('\n'.join(str(f(int(l))) for l in file('test.txt', 'r').readlines())) 


# And a wrapper for sanity 
def filter_file(infile, outfile, filter_function) 
    outfile.write('\n'.join(filter_function(l) for l in infile.readlines()))