How do I create a temporary directory in Python?

Additional ways to control and secure temporary directories

Yes, there's more! With Python, you have additional ways to create temporary directories, from handling manual cleanup to ensuring essential security.

Manual cleanup: The choice is yours

If you want to do a manual cleanup, here’s how to use tempfile.mkdtemp() and shutil.rmtree():

import tempfile
import shutil

temp_dir = tempfile.mkdtemp()
print('Temp dir:', temp_dir)

# Do things in the temp directory

# Cleanup when you decide, not when Python decides
shutil.rmtree(temp_dir)


Nice, isn’t it? Now you call the shots here, not Python!

Security: Yep, we’ve got that covered too!

Use tempfile.mkdtemp() to securely create a temp directory and use shutil.rmtree() for removing it. But remember, exceptions are the party crashers. So, handle them:

import tempfile
import shutil
import os

try:
    temp_dir = tempfile.mkdtemp()
    # Perform operations in temp_dir
finally:
    shutil.rmtree(temp_dir)  # Cleanup crew to the rescue


Switching directories: Home is where your temp_dir is

With os.chdir(), switch to your temporary directory within the context manager:

import tempfile
import os

with tempfile.TemporaryDirectory() as temp_dir:
    os.chdir(temp_dir)
    # Now we're cooking in temp_dir kitchen


And no dirty dishes left behind! The context manager will clean everything up.

Navigating advanced usage and error handling

You're not limited to the basics. Python offers advanced usage options with contextlib, pathlib, and even preventive error handling.

Making your own context manager

Want to roll your context manager? You can with contextlib. It works together with tempfile.mkdtemp() for flexibility:

import os
import tempfile
from contextlib import contextmanager
import shutil

@contextmanager
def custom_temp_directory():
    dir_path = tempfile.mkdtemp()
    try:
        yield dir_path
    finally:
        shutil.rmtree(dir_path)

with custom_temp_directory() as temp_dir:
    # Party in the custom temp_dir


Now you're not just a Pythonista, you're a maestro!

Using pathlib for paths

The pathlib module comes as a new, cool kid on the block for handling filesystem paths. So, meet your new friend:

from pathlib import Path
import tempfile

with tempfile.TemporaryDirectory() as temp_dir:
    temp_path = Path(temp_dir)
    # Meet the power of slash operator
    file_path = temp_path / 'myfile.txt'
    # Continue operations here


Pathlib makes Python coding feel like a cool music jamming session. 🎸

Preventing ResourceWarning: Safety first!

Let's put on our safety helmets here and prevent ResourceWarning in Python's unittest with proper cleanup in the teardown methods:

import unittest
import tempfile

class MyTestCase(unittest.TestCase):
    def setUp(self):
        self.temp_dir = tempfile.TemporaryDirectory()

    def tearDown(self):
        self.temp_dir.cleanup()  # Cleanup crew at your service!

    def test_something(self):
        # Test something here
