วิธีสร้างไดเรกทอรีชั่วคราวและรับชื่อพา ธ / ไฟล์ใน Python

วิธีสร้างไดเรกทอรีชั่วคราวและรับชื่อพา ธ / ไฟล์ในหลาม

ใช้mkdtemp()ฟังก์ชั่นจากtempfileโมดูล:

import tempfile
import shutil

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

ใน Python 3 สามารถใช้TemporaryDirectoryในโมดูลtempfileได้

ตรงจากตัวอย่าง :

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()วิธีการ"

หากต้องการขยายคำตอบอื่น ๆ นี่เป็นตัวอย่างที่ค่อนข้างสมบูรณ์ซึ่งสามารถล้าง 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 และใหม่กว่ามี contextmanager ที่มีประโยชน์สำหรับสิ่งนี้ใน 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)