18 lines
269 B
Python
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)
|
||
|
|
|
||
|
|
|