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


当前回答

在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: implicit cleanup up <TemporaryDirectory…如果你依赖这个。

其他回答

使用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中,可以使用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: implicit cleanup up <TemporaryDirectory…如果你依赖这个。

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

如果我正确地回答了您的问题,您还想知道在临时目录中生成的文件的名称吗? 如果是,试试这个:

import os
import tempfile

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