programing

파이썬에서 임시 디렉터리를 만들려면 어떻게 해야 합니까?

i4 2023. 6. 10. 08:18
반응형

파이썬에서 임시 디렉터리를 만들려면 어떻게 해야 합니까?

파이썬에서 임시 디렉터리를 만들고 경로를 얻으려면 어떻게 해야 합니까?

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

예제에서:

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()

설명서에는 다음과 같은 내용도 나와 있습니다.

컨텍스트가 완료되거나 임시 디렉터리 개체가 삭제되면 새로 생성된 임시 디렉터리와 해당 디렉터리의 모든 내용이 파일 시스템에서 제거됩니다.

예를 들어 프로그램이 끝날 때 Python은 디렉토리가 제거되지 않은 경우(예: 컨텍스트 관리자 또는cleanup()방법.파이썬의unittest에 대해 불평할 수도 있습니다.ResourceWarning: Implicitly cleaning up <TemporaryDirectory...하지만 당신이 이것에 의지한다면요.

모듈의 기능을 사용합니다.

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 에 이에 대한 유용한 컨텍스트 관리자가 있습니다.

문서에서는 임시 디렉토리를 만들고 컨텍스트 관리자를 종료할 때 자동으로 제거하는 컨텍스트 관리자를 사용할 것을 제안합니다.

import tempfile

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)
    file_name = temp_dir / "test.txt"
    file_name.write_text("bla bla bla")

    print(temp_dir, temp_dir.exists())
    # /tmp/tmp81iox6s2 True

    print(file_name, "contains", file_name.open().read())
    # /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

질문이 맞다면 임시 디렉토리 내에서 생성된 파일의 이름도 알고 싶으십니까?그렇다면 다음을 시도해 보십시오.

import os
import tempfile

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

언급URL : https://stackoverflow.com/questions/3223604/how-do-i-create-a-temporary-directory-in-python

반응형