Trong python, tạo một tempfile trong cùng thư mục với một tệp khác?

Question

Tôi cần cập nhật tệp. Tôi đọc nó và viết nó ra với những thay đổi. Tuy nhiên, tôi muốn viết vào một tập tin tạm thời và đổi tên nó thành vị trí.

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

Vấn đề là tempfile.NamedTemporaryFile() tạo tệp tạm thời trong /tmp là hệ thống tệp khác. Điều này có nghĩa là os.rename() không thành công. Nếu tôi sử dụng shlib.move() thay vào đó thì tôi không có bản cập nhật nguyên tử "mv" cung cấp (cho các tệp trong cùng hệ thống tệp, yadda, yadda, v.v.)

Tôi biết tempfile.NamedTemporaryFile() có thông số "dir", nhưng data_file_name có thể là "foo.txt" trong trường hợp dir = '.'; hoặc data_file_name có thể là "/path/to/the/data/foo.txt" trong trường hợp đó dir = "/path/to/the/data".

Điều tôi thực sự muốn là tệp tạm thời là data_file_name + "một số dữ liệu ngẫu nhiên". Điều này sẽ có lợi ích của thất bại trong một cách mà sẽ để lại đằng sau những manh mối hữu ích.

Đề xuất?

Answer 1

Bạn có thể sử dụng:

• prefix để làm cho các tập tin tạm thời bắt đầu với tên giống như file gốc.
• dir để chỉ định vị trí đặt tệp tạm thời.
• os.path.split để chia thư mục khỏi tên tệp.

---

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

Answer 2

tôi sử dụng thời gian hiện tại là "một số dữ liệu ngẫu nhiên" nối vào một chuỗi cơ sở cho một tên tập tin tạm thời duy nhất:

import time 

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

Answer 3

Bạn có thể vượt qua một vị trí tập tin trong tham số constructor 'dir'. Nó hoạt động, như bạn muốn.

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

Nguồn: http://docs.python.org/library/tempfile.html#tempfile.NamedTemporaryFile

Answer 4

Để đáp ứng tất cả các danh sách kiểm tra của bạn tôi nghĩ rằng bạn muốn sử dụng ...

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

quan trọng để có delete=False, bởi vì nếu không:

[...] Nếu xóa là đúng (mặc định), tệp sẽ bị xóa ngay sau khi được đóng .

Answer 5

Mô-đun tempfile mà bạn sử dụng cung cấp cách quản lý tệp tạm thời an toàn. Nếu bạn thực sự muốn sử dụng hệ thống của riêng mình, bạn nên lưu ý rằng nó có thể dễ bị tấn công (đặc biệt là các cuộc tấn công của symlink).

Một cách đơn giản để tạo ra một tên tập tin độc đáo tạm thời (mặc dù một tên khá dài) là:

import uuid 
import os 

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

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

os.remove(tempfilename) 

Nhưng đây là một chút hackish; thực sự thay vì xem xét sử dụng mô-đun tempfile với các thông số prefix và dir chính xác được chuyển đến NamedTemporaryFile, như được mô tả trong các câu trả lời khác.