vagrant_fabric.py

from fabric.api import *
import vagrant

def with_vagrant(vm_name=None):
    """Apply the rule with the vagrant box."""
    print("applying on box {}".format(vm_name))
    v = vagrant.Vagrant()
    if not all([status == "running" for status in v.status().values()]):
        print("Vagrant is not running, starting it right now.")
        v.up()

    while not all([status == "running" for status in v.status().values()]):
        print("Wait for vagrant to be up.")
        time.sleep(1)

    v_conf = v.conf(vm_name=vm_name)
    ssh_key = v_conf.get('IdentityFile')
    ssh_string = v.user_hostname_port(vm_name=vm_name)
    env.hosts.append(ssh_string)
    #env.host = ssh_string
    env.key_filename = ssh_key