如何创建一个临时目录，并获得其路径/文件名?

我如何在Python中创建一个临时目录并获取其路径/文件名?

219020 次浏览
小开

使用tempfile模块中的mkdtemp()函数:

import tempfile
import shutil


dirpath = tempfile.mkdtemp()
# ... do stuff with dirpath
shutil.rmtree(dirpath)
小开

为了扩展另一个答案，这里有一个相当完整的示例，它可以在异常时清理tmpdir:

import contextlib
import os
import shutil
import tempfile


@contextlib.contextmanager
def cd(newdir, cleanup=lambda: True):
prevdir = os.getcwd()
os.chdir(os.path.expanduser(newdir))
try:
yield
finally:
os.chdir(prevdir)
cleanup()


@contextlib.contextmanager
def tempdir():
dirpath = tempfile.mkdtemp()
def cleanup():
shutil.rmtree(dirpath)
with cd(dirpath, cleanup):
yield dirpath


def main():
with tempdir() as dirpath:
pass # do something here
小开

在python 3.2及以后版本中，在stdlib https://docs.python.org/3/library/tempfile.html#tempfile.TemporaryDirectory中有一个有用的上下文管理器

小开

在Python 3中，可以使用tempfile模块中的TemporaryDirectory。

从例子:

import tempfile


with tempfile.TemporaryDirectory() as tmpdirname:
print('created temporary directory', tmpdirname)


# directory and contents have been removed

要手动控制何时删除目录，不要使用上下文管理器，如下例所示:

import tempfile


temp_dir = tempfile.TemporaryDirectory()
print(temp_dir.name)
# use temp_dir, and when done:
temp_dir.cleanup()

文档还说:

在完成上下文或销毁临时目录对象时，新创建的临时目录及其所有内容将从文件系统中删除。

例如，在程序结束时，如果目录没有被上下文管理器或cleanup()方法删除，Python将清理该目录。不过，如果你依赖于此，Python的unittest可能会抱怨ResourceWarning: Implicitly cleaning up <TemporaryDirectory...。

小开
如果我正确地得到了你的问题，你还想知道在临时目录中生成的文件的名称? 如果是，试试这个:

import os
import tempfile


with tempfile.TemporaryDirectory() as tmp_dir:
# generate some random files in it
files_in_dir = os.listdir(tmp_dir)
小开

Docs.python.org提供一个临时目录 使用实例 上下文管理器:< / p > 

import tempfile
# create a temporary directory using the context manager
with tempfile.TemporaryDirectory() as tmpdirname:
print('created temporary directory', tmpdirname)


# Outside the context manager, directory and contents have been removed.

使用pathlib来方便在tempfile之上的路径操作，使得使用pathlib的/路径操作符创建新路径成为可能:

import tempfile
from pathlib import Path
with tempfile.TemporaryDirectory() as tmpdirname:
temp_dir = Path(tmpdirname)
print(temp_dir, temp_dir.exists())
file_name = temp_dir / "test.txt"
file_name.write_text("bla bla bla")
print(file_name, "contains", file_name.open().read())


# /tmp/tmp81iox6s2 True
# /tmp/tmp81iox6s2/test.txt contains bla bla bla

在上下文管理器之外，文件已被销毁

print(temp_dir, temp_dir.exists())
# /tmp/tmp81iox6s2 False
print(file_name, file_name.exists())
# /tmp/tmp81iox6s2/test.txt False

提交答案