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)