Change directory context (python)

CD.py

class cd:
    """
    Context manager for changing the current working directory
    
    with cd('/path'):
      open('file_in_path')
    """
    def __init__(self, newPath):
        self.newPath = newPath

    def __enter__(self):
        self.savedPath = os.getcwd()
        os.chdir(self.newPath)

    def __exit__(self, etype, value, traceback):
        os.chdir(self.savedPath)

Functional version:

import os
import contextlib

@contextlib.contextmanager
def cd(new_path):
    old_cwd = os.getcwd()
    os.chdir(new_path)
    yield
    os.chdir(old_cwd)

Note that in Python3.11 this function is included as contextlib.chdir.