sshcm.py

gistfile1.txt

from fabric import network as net

@contextmanager
def ssh2(notify, raise_error=False):
    try:
        yield 
    except :
        exc_t, exc, tb = sys.exc_info()
        if isinstance(exc, exceptions.SystemExit):
            notify('Task aborted. Check fabworker logs.')
        else:
            notify('Task error: %s' %(exc))
            logger.error(traceback.format_exc())
        if raise_error:
            raise
    finally:
        net.disconnect_all()