Mientras trabajaba en un script completamente diferente, aprendí que con 29 millones de líneas de texto, el uso seek()
y la operación de byte de datos a menudo es más rápido que línea por línea. La misma idea se aplica en la secuencia de comandos a continuación: abrimos el archivo y, en lugar de recorrerlo abriendo y cerrando el archivo (lo que puede agregar una sobrecarga, incluso si no es significativo), mantenemos el archivo abierto y volvemos al principio.
#!/usr/bin/env python3
from __future__ import print_function
import sys,os
def error_out(string):
sys.stderr.write(string+"\n")
sys.exit(1)
def read_bytewise(fp):
data = fp.read(1024)
print(data.decode(),end="",flush=True)
while data:
data = fp.read(1024)
print(data.decode(),end="",flush=True)
#fp.seek(0,1)
def main():
howmany = int(sys.argv[1]) + 1
if not os.path.isfile(sys.argv[2]):
error_out("Needs a valid file")
fp = open(sys.argv[2],'rb')
for i in range(1,howmany):
#print(i)
fp.seek(0)
read_bytewise(fp)
fp.close()
if __name__ == '__main__': main()
El script en sí es bastante simple en uso:
./repeat_text.py <INT> <TEXT.txt>
Para un archivo de texto de 3 líneas y 1000 iteraciones, todo va bastante bien, aproximadamente 0.1 segundos:
$ /usr/bin/time ./repeat_text.py 1000 input.txt > /dev/null
0.10user 0.00system 0:00.23elapsed 45%CPU (0avgtext+0avgdata 9172maxresident)k
0inputs+0outputs (0major+1033minor)pagefaults 0swaps
El script en sí no es muy elegante, probablemente podría acortarse, pero hace el trabajo. Por supuesto, agregué algunos bits adicionales aquí y allá, como la error_out()
función, que no es necesaria, es solo un pequeño toque fácil de usar.