使用文件输出自动创建目录[重复]

Question

可能重复：
mkdir -p functionality in python

假设我要制作一个文件：

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

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

这给出了IOError，因为/foo/bar 不存在。

自动生成这些目录的最 Pythonic 方式是什么？我是否有必要在每一个上显式调用 os.path.exists 和 os.mkdir（即 /foo，然后 /foo/bar）？

Answer 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+ 中，有一个 more elegant way 可以避免上述竞争条件：

import os

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