restoring cwd with a decorator.

gistfile1.py

def restore_cwd(func):
    """ 
    decorator to restore the current working directory
    """
    def wrapper(*args, **kwargs):
        cwd = os.getcwd()
        results = func(*args, **kwargs)
        os.chdir(cwd)
        return results

    return wrapper