Автоматическое создание каталогов с выходом файла

Возможный дубликат:
Функции mkdir -p в python

Скажем, я хочу сделать файл:

filename = "/foo/bar/baz.txt"

with open(filename, "w") as f:
    f.write("FOOBAR")

Это дает IOError, так как /foo/bar не существует.

Каков самый пифонический способ автоматического создания этих каталогов? Нужно ли мне явно называть os.path.exists и os.mkdir на каждом из них (т.е./Foo, then/foo/bar)?

Ответы

Ответ 1

Функция os.makedirs делает это. Попробуйте следующее:

import os
import errno

filename = "/foo/bar/baz.txt"
if not os.path.exists(os.path.dirname(filename)):
    try:
        os.makedirs(os.path.dirname(filename))
    except OSError as exc: # Guard against race condition
        if exc.errno != errno.EEXIST:
            raise

with open(filename, "w") as f:
    f.write("FOOBAR")

Причиной добавления блока try-except является обработка случая, когда каталог был создан между вызовами os.path.exists и os.makedirs, чтобы защитить нас от условий гонки.

В Python 3.2+ существует более более элегантный способ, который позволяет избежать состояния гонки выше:

filename = "/foo/bar/baz.txt"¨
os.makedirs(os.path.dirname(filename), exist_ok=True)
with open(filename, "w") as f:
    f.write("FOOBAR")