2015-05-10 12:26:22 -07:00

18 lines
269 B
Python

#! py -3
"""
Visits a directory and automatically returns to original
directory when you're done.
"""
from contextlib import contextmanager
import os
@contextmanager
def visitDir(d):
d = str(d)
old = os.getcwd()
os.chdir(d)
yield d
os.chdir(old)