21 lines
373 B
Python
21 lines
373 B
Python
|
import glob
|
||
|
import os
|
||
|
import subprocess
|
||
|
import traceback
|
||
|
|
||
|
from contextlib import contextmanager
|
||
|
|
||
|
@contextmanager
|
||
|
def pushd(new_dir):
|
||
|
previous_dir = os.getcwd()
|
||
|
os.chdir(new_dir)
|
||
|
try:
|
||
|
yield
|
||
|
finally:
|
||
|
os.chdir(previous_dir)
|
||
|
|
||
|
|
||
|
def removeall(dir):
|
||
|
[os.remove(f) for f in glob.iglob("{}/*".format(dir), recursive=True)]
|
||
|
os.removedirs(dir)
|