python pushd

pushd.py

from contextlib import contextmanager
import os


@contextmanager
def pushd(new_dir):
    prev_dir = os.getcwd()
    os.chdir(new_dir)
    yield
    os.chdir(prev_dir)


if __name__ == "__main__":
    with pushd('./my_dir'):
        print(os.getcwd())  # ./my_dir
        
        with pushd('./my_dir/my_other_dir'):
            print(os.getcwd())  # ./my_dir/my_other_dir
        
        print(os.getcwd())  # ./my_dir