使用文件输出自动创建目录

时间:2012-09-20 17:06:41

标签: python file-io

  

可能重复:
  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.existsos.mkdir(即/ foo,然后是/ foo / bar)?

1 个答案:

答案 0 :(得分:507)

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.existsos.makedirs调用之间创建目录的情况,以便保护我们免受竞争条件的影响。< / p>


在Python 3.2+中,有一个more elegant way可以避免上述竞争条件:

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