En python, crea un archivo temporal en el mismo directorio que otro archivo?

Question

Necesito actualizar un archivo. Lo leí y lo escribí con los cambios. Sin embargo, preferiría escribir en un archivo temporal y cambiarle el nombre.

temp = tempfile.NamedTemporaryFile() 
tempname = temp.name 
temp.write(new_data) 
temp.close() 
os.rename(tempname, data_file_name) 

El problema es que tempfile.NamedTemporaryFile() hace que el archivo temporal en /tmp que es un sistema de archivos diferente. Esto significa que os.rename() falla. Si uso shlib.move() entonces no tengo la actualización atómica que "mv" ofrece (para archivos en el mismo sistema de archivos, bla, bla, etc.)

Sé tempfile.NamedTemporaryFile() toma un parámetro "dir", pero data_file_name podría ser "foo.txt" en cuyo caso dir = '.'; o data_file_name podría ser "/path/to/the/data/foo.txt" en cuyo caso dir = "/path/to/the/data".

Lo que realmente me gustaría es que el archivo temporal sea data_file_name + "algunos datos aleatorios". Esto tendría el beneficio de fallar de una manera que dejaría pistas útiles.

Sugerencias?

Answer 1

Usted puede utilizar:

• prefix para hacer que el archivo temporal comienzan con el mismo nombre que el archivo original .
• dir para especificar dónde colocar el archivo temporal.
• os.path.split para dividir el directorio del nombre de archivo.

---

import tempfile 
import os 
dirname, basename = os.path.split(filename) 
temp = tempfile.NamedTemporaryFile(prefix=basename, dir=dirname) 
print(temp.name) 

Answer 2

utilizo el momento actual como "algunos datos aleatorios" anexan a una cadena de base para un nombre de archivo temporal único:

import time 

temp_file_name = data_file_name + str(time.time()) 

Answer 3

Puede pasar una ubicación de archivo en el parámetro constructor 'dir'. Funciona, como lo desee.

>>> t = tempfile.NamedTemporaryFile(dir="/Users/rafal") 
>>> t.name 
'/Users/rafal/tmplo45Js' 

Fuente: http://docs.python.org/library/tempfile.html#tempfile.NamedTemporaryFile

Answer 4

para satisfacer todas sus lista de verificación Creo que te gustaría utilizar ...

temp = tempfile.NamedTemporaryFile(prefix=data_file_name, dir=path, 
            delete=False) 

importante tener la delete=False, porque de lo contrario:

[...] Si la eliminación es verdadera (la predeterminada), el archivo se elimina tan pronto como se cierra .

Answer 5

El módulo tempfile que utiliza proporciona una forma segura de administrar los archivos temporales. Si realmente quieres usar tu propio sistema, debes ser consciente de que podría ser vulnerable a ataques (particularmente ataques de enlace simbólico).

Una forma sencilla de generar un nombre de archivo único temporal (aunque un nombre bastante largo) es:

import uuid 
import os 

tempfilename = 'myprefix-%s.dat' % str(uuid.uuid4()) 

with open(tempfilename, 'rw') as tempfile: 
    # do stuff 

os.remove(tempfilename) 

Pero esto es un poco hacker; realmente más bien considere usar el módulo tempfile con los parámetros prefix y dir correctos pasados ​​a NamedTemporaryFile, como se describe en las otras respuestas.