파이썬에서 임시 디렉토리를 만들고 경로 / 파일 이름을 얻는 방법


답변:


210

모듈 의 mkdtemp()기능을 사용하십시오 tempfile.

import tempfile
import shutil

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

7
테스트에서이 파일을 사용하는 경우 사용 후 디렉토리가 자동으로 삭제되지 않으므로 디렉토리 (shutil.rmtree)를 제거하십시오. "mkdtemp ()의 사용자는 임시 디렉토리와 디렉토리 디렉토리의 내용을 삭제해야합니다." 참조 : docs.python.org/2/library/tempfile.html#tempfile.mkdtemp
닐스 그랬지

97
python3에서는 할 수 with tempfile.TemporaryDirectory() as dirpath:있으며 컨텍스트 관리자를 종료하면 임시 디렉토리가 자동으로 정리됩니다. docs.python.org/3.4/library/…
대칭

41

Python 3에서는 tempfile 모듈의 TemporaryDirectory 를 사용할 수 있습니다.

이것은 예제와 직결됩니다 .

import tempfile
with tempfile.TemporaryDirectory() as tmpdirname:
     print('created temporary directory', tmpdirname)
# directory and contents have been removed

디렉토리를 조금 더 길게 유지하려면 다음과 같이 할 수 있습니다 (예가 아닌).

import tempfile
import shutil

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

@MatthiasRoelandts가 지적했듯이 문서는 또한 " cleanup()메소드 를 호출하여 디렉토리를 명시 적으로 정리할 수있다"고 말합니다 .


2
shutil.rmtree (temp_dir.name)는 필요하지 않습니다.
sidcha

37

다른 답변을 확장하기 위해 예외 상황에서도 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


3

질문이 올바르게 나타나면 임시 디렉토리 내에 생성 된 파일의 이름도 알고 싶습니까? 그렇다면 다음을 시도하십시오.

import os
import tempfile

with tempfile.TemporaryDirectory() as tmp_dir:
    # generate some random files in it
     files_in_dir = os.listdir(tmp_dir)
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.