W pythonie, należy utworzyć plik tempfile w tym samym katalogu, co inny plik?

Question

Potrzebuję zaktualizować plik. Przeczytałem to i zapisałem ze zmianami. Wolę jednak pisać do pliku tymczasowego i zmienić jego nazwę na miejsce.

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

Problemem jest to, że tempfile.NamedTemporaryFile() sprawia plik tymczasowy w /tmp który jest inny system plików. Oznacza to, że nie powiodło się os.rename(). Jeśli użyję zamiast tego shlib.move(), to nie mam aktualizacji atomowej, którą zapewnia "mv" (dla plików w tym samym systemie plików, Yadda, Yadda itp.).

Wiem, że tempfile.NamedTemporaryFile() przyjmuje parametr "dir", ale data_file_name może być "foo.txt", w takim przypadku dir = '.'; lub data_file_name może być "/path/to/the/data/foo.txt", w takim przypadku dir = "/path/to/the/data".

Co naprawdę chciałbym to plik tymczasowy to data_file_name + "kilka losowych danych". Miałoby to tę zaletę, że zawiodło w sposób, który pozostawiłby po sobie przydatne wskazówki.

Sugestie?

Answer 1

Można użyć:

• prefix aby rozpocząć tymczasowy plik o tej samej nazwie co plik oryginalny .
• dir aby określić, gdzie umieścić plik tymczasowy.
• os.path.split, aby podzielić katalog z nazwą pliku.

---

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

Answer 2

używam aktualny czas jako „pewne losowe dane” dołączany do łańcucha bazowej dla unikalnej nazwy pliku tymczasowego:

import time 

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

Answer 3

można przekazać lokalizację pliku w „dir” parametr konstruktora. Działa, jak chcesz.

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

Źródło: http://docs.python.org/library/tempfile.html#tempfile.NamedTemporaryFile

Answer 4

aby spełnić wszystkie Państwa kontrolną myślę, że chcesz korzystać z ...

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

Ważne, aby mieć delete=False, bo inaczej:

[...] Jeśli delete jest prawdziwe (domyślne), plik zostanie usunięty, gdy tylko zostanie zamknięty .

Answer 5

Używany przez Ciebie moduł tempfile zapewnia bezpieczny sposób zarządzania plikami tymczasowymi. Jeśli naprawdę chcesz korzystać z własnego systemu, powinieneś pamiętać, że może on być podatny na ataki (w szczególności ataki z użyciem dowiązań symbolicznych).

Prosty sposób, aby wygenerować tymczasowej unikalną nazwę pliku (choć dość długą nazwą) jest:

import uuid 
import os 

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

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

os.remove(tempfilename) 

Ale to jest nieco hackish; Naprawdę warto rozważyć użycie modułu tempfile z poprawnymi parametrami prefix i dir przekazanymi do NamedTemporaryFile, jak opisano w innych odpowiedziach.