Cómo creo un archivo en python sin sobreescribir un archivo existente

Question

Actualmente tengo un bucle que intenta encontrar un nombre de archivo sin usar agregando sufijos a una cadena de nombre de archivo. Una vez que no puede encontrar un archivo, usa el nombre que no pudo abrir un nuevo archivo con ese nombre. El problema es que este código se usa en un sitio web y puede haber múltiples intentos de hacer lo mismo al mismo tiempo, por lo que existe una condición de carrera.

¿Cómo puedo evitar que python sobrescriba un archivo existente, si se crea entre el momento de la comprobación y el momento de la apertura en el otro hilo?

Puedo minimizar la posibilidad de aleatorizar los sufijos, pero la posibilidad ya se ha minimizado en función de las partes de la ruta de acceso. Quiero eliminar esa posibilidad con una función que se pueda contar, crear este archivo SOLAMENTE si no existe.

Puedo usar las funciones de win32 para hacer esto, pero quiero que esto funcione en plataforma cruzada porque al final se alojará en Linux.

Answer 1

Uso os.open() con os.O_CREAT y os.O_EXCL para crear el archivo. Que pueden fallar si el archivo ya existe:

>>> fd = os.open("x", os.O_WRONLY | os.O_CREAT | os.O_EXCL) 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
OSError: [Errno 17] File exists: 'x' 

Una vez que haya creado un nuevo archivo, utilice os.fdopen() para girar el mango en un objeto de archivo estándar de Python:

>>> fd = os.open("y", os.O_WRONLY | os.O_CREAT | os.O_EXCL) 
>>> f = os.fdopen(fd, "w") # f is now a standard Python file object 

Editar: De Python 3.3, el built-in open() tiene un modo x que significa "abierto para creación exclusiva, fallando si el archivo ya existe".

Answer 2

Si le preocupa una condición de carrera, puede create a temporary file y luego cambiarle el nombre.

>>> import os 
>>> import tempfile 
>>> f = tempfile.NamedTemporaryFile(delete=False) 
>>> f.name 
'c:\\users\\hughdb~1\\appdata\\local\\temp\\tmpsmdl53' 
>>> f.write("Hello world") 
>>> f.close() 
>>> os.rename(f.name, r'C:\foo.txt') 
>>> if os.path.exists(r'C:\foo.txt') : 
...  print 'File exists' 
... 
File exists 

Como alternativa, puede crear los archivos con un uuid en el nombre. Stackoverflow artículo en esto.

>>> import uuid 
>>> str(uuid.uuid1()) 
'64362370-93ef-11de-bf06-0023ae0b04b8' 

Answer 3

Si tiene un id asociado a cada subproceso/proceso que intenta crear el archivo, puede poner ese id en el sufijo en alguna parte, garantizando así que no hay dos procesos que puedan usar el mismo nombre de archivo.

Esto elimina la condición de carrera entre los procesos.