cách tạo tệp và ném ngoại lệ nếu đã tồn tại

Question

Trong chương trình của tôi, nhiều quy trình có thể thử tạo tệp nếu tệp không tồn tại. Bây giờ tôi muốn đảm bảo rằng chỉ một trong các quy trình có thể tạo tệp và phần còn lại nhận được một ngoại lệ nếu nó đã được tạo (loại quy trình an toàn và mở thread an toàn). Làm thế nào tôi có thể đạt được điều này trong python.

Chỉ cần cho rõ ràng, những gì tôi muốn là tập tin được tạo ra nếu nó không tồn tại. Nhưng nếu nó đã tồn tại rồi ném một ngoại lệ. Và tất cả điều này sẽ xảy ra một cách nguyên tử.

Answer 1

Trong Python 2.x:

import os 

fd = os.open('filename', os.O_CREAT|os.O_EXCL) 
with os.fdopen(fd, 'w') as f: 
    .... 

Trong Python 3.3+:

with open('filename', 'x') as f: 
    .... 

Answer 2

Nếu bạn đang chạy trên một hệ thống Unix-like, mở tập tin như thế này:

f = os.fdopen(os.open(filename, os.O_CREAT | os.O_WRONLY | os.O_EXCL), 'w') 

Các O_EXCL cờ để os.open đảm bảo rằng các tập tin sẽ chỉ được tạo ra (và mở) nếu nó doesn' đã tồn tại, nếu không, ngoại lệ OSError sẽ được nâng lên. Việc kiểm tra sự tồn tại và tạo ra sẽ được thực hiện một cách nguyên tử, vì vậy bạn có thể có nhiều luồng hoặc các tiến trình cạnh tranh để tạo ra tệp, và chỉ có một cái sẽ thành công.