fabric vagrant integration

fabfile.py

# -*- coding: utf-8 -*-

from fabric.api import env
from fabric.decorators import task
from fabric.operations import run, local

env.timeout = 120
# env.hosts = ['motoko', 'batou', 'tachikoma']
env.forward_agent = True
env.use_ssh_config = True


@task(default=True)
def uname():
    run('uname -a')


@task
def vagrant():
    output = local('vagrant ssh-config', capture=True)
    sshcfg = dict(a.split() for a in output.split('\n'))
    env.user = sshcfg['User']
    env.port = sshcfg['Port']
    env.hosts = [sshcfg['HostName']]
    env.key_filename = sshcfg['IdentityFile'].replace('"', '')