from fabric.api import *
from fabric.context_managers import settings
from fabric.decorators import roles
from fabric.api import run, cd, env, roles
from fabric.contrib.files import exists,upload_template
from pprint import pprint
from time import sleep, time
import yaml
import fabric
import requests

vuln_hosts = []
patched_hosts = []
env.use_ssh_config = True
config_dir = '/home/sam/work/wyse/config/'

env.roledefs = {
        'so1': [],
        'sd1': [],
        'sq1': [],
        'sd2': [],
        'sq2': [],
        'ss1': [],
        'ss2': [],
        'sb1': [],
        'sb2': []
        }

# Loads an environment from a yaml config and stores in env.config
def loadenv(environment = ''):
    with open(config_dir + environment + '.yaml', 'r') as f:
        env.config = yaml.load(f)
        env.password = env.config['password']
        env.roledefs = env.config['roledefs']
        env.username = env.config['username']