In python, creare un tempfile nella stessa directory di un altro file?

Question

Ho bisogno di aggiornare un file. L'ho letto e lo ho scritto con le modifiche. Tuttavia, preferirei scrivere su un file temporaneo e rinominarlo in posizione.

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

Il problema è che tempfile.NamedTemporaryFile() rende il file temporaneo nella /tmp che è un file system diverso. Ciò significa che os.rename() non riesce. Se uso shlib.move() invece poi non ho l'aggiornamento atomica che "mv" prevede (per i file nello stesso file system, bla, bla, ecc)

So tempfile.NamedTemporaryFile() prende un parametro "dir", ma data_file_name potrebbe essere "foo.txt" nel qual caso dir = '.'; o data_file_name potrebbe essere "/path/to/the/data/foo.txt" nel qual caso dir = "/path/to/the/data".

Quello che mi piacerebbe davvero è che il file temp sia data_file_name + "alcuni dati casuali". Ciò avrebbe il vantaggio di fallire in un modo che lascerebbe indizi utili.

Suggerimenti?

Answer 1

È possibile utilizzare:

• prefix per rendere il file temporaneo iniziano con lo stesso nome del file originale .
• dir per specificare dove posizionare il file temporaneo.
• os.path.split per dividere la directory dal nome file.

---

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

Answer 2

Io uso l'ora attuale come "alcuni dati casuali" aggiunti a una stringa di base per un nome di file temporaneo unico:

import time 

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

Answer 3

È possibile passare un percorso di file nel parametro del costruttore 'dir'. Funziona, come desideri.

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

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

Answer 4

Per soddisfare tutte le vostre lista di controllo penso che ci si vuole utilizzare ...

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

importante avere il delete=False, perché altrimenti:

[...] Se delete è true (il valore predefinito), il file viene eliminato non appena è chiuso.

Answer 5

Il modulo tempfile che si utilizza fornisce un modo sicuro di gestione dei file temporanei. Se si vuole veramente utilizzare il proprio sistema, si dovrebbe essere consapevoli che potrebbe essere vulnerabile agli attacchi (in particolare gli attacchi symlink).

Un modo semplice per generare un nome di file unico temporaneo (anche se un nome piuttosto lungo) è:

import uuid 
import os 

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

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

os.remove(tempfilename) 

Ma questo è un po 'hacker; considerare piuttosto piuttosto l'utilizzo del modulo tempfile con i parametri corretti prefix e dir passati a NamedTemporaryFile, come descritto nelle altre risposte.