run_in_namespace.py

def run_in_namespace(namespace):
    """
    Decorator for executing a function in a given namespace, then return back to the
    current namescape.
    """
    def decorator(function):
        def wrapper(*args, **kwargs):
            current_namespace = namespace_manager.get_namespace()
            try:
                current_namespace = namespace_manager.get_namespace()
                namespace_manager.set_namespace(namespace)
                return function(*args, **kwargs)
            finally:
                namespace_manager.set_namespace(current_namespace)
        return wrapper
    return decorator