pythonでcdしたい

cd.py

import contextlib
import os


@contextlib.contextmanager
def cd(path):
    current_path = os.getcwd()
    os.chdir(path)
    try:
        yield
    finally:
        os.chdir(current_path)

example.py

import os
from cd import cd


def main():
    print(os.getcwd())
    with cd("/path/you/wanna/go"):
        print(os.getcwd())
    print(os.getcwd())
    
    
if __name__ == "__main__":
    main()