Python pushd implementation using context manager

pushd.py

import os
from contextlib import contextmanager

@contextmanager
def pushd(newdir):
    old_path = os.getcwd()
    os.chdir(newdir)
    try:
        yield
    finally:
        os.chdir(old_path)