def __init__(self):
self._config = None
# Load config
self.load_config()
# store master ip
self._docean = DOcean(self._config['api-key'])
# google drive
if self._config['google-drive']:
self._google_drive_session = 'google-session.json'
self._google_drive = GoogleDriveAPIWrapper('',self._google_drive_session)
class ElasticPowerTAC_Slave:
# constructor
def __init__(self):
self._config = None
# Load config
self.load_config()
# store master ip
self._docean = DOcean(self._config['api-key'])
# google drive
if self._config['google-drive']:
self._google_drive_session = 'google-session.json'
self._google_drive = GoogleDriveAPIWrapper('',self._google_drive_session)
# load_config
def load_config(self):
# load from "config.json"
try:
config_file = "config.json"
self._config = None
with open(config_file, 'r') as f:
self._config = f.read()
self._config = json.loads(self._config)
except:
print('config.json must be defined.')
exit()
# setup slave environment
def setup_slave_simulations(self):
print("Slaves have been initialized!")
self.setup_scenarios()
# start simulation scenarios
def start_slave_simulations(self):
# start simulation runner
os.chdir('/home/log/ElasticPowerTAC-Simulation')
run_cmd = ['su', 'log', '-c', 'python simulation.py']
subprocess.call(run_cmd)
os.chdir('/root/ElasticPowerTAC-Slave')
# simulations are complete by this point
if self._config['google-drive']:
self.backup_on_google_drive()
# backup on google drive
def backup_on_google_drive(self):
# iterate through files in simulation location and upload tar.gz files
path = '/home/log/ElasticPowerTAC-Simulation'
for filename in os.listdir(path):
if filename.find('tar.gz')>=0:
self._google_drive.insert_file(filename,
filename,
self._config['google-drive']['parent-id'],
'application/x-gzip',
'%s/%s'%(path,filename))
# move scenarios from root to log location
def setup_scenarios(self):
# Location path
run_mv = ['mv', '/root/ElasticPowerTAC-Slave/scenarios', '/home/log/ElasticPowerTAC-Simulation/']
subprocess.call(run_mv)
run_chown = ['chown','log:log','/home/log/ElasticPowerTAC-Simulation/scenarios']
subprocess.call(run_chown)
# destroy slave :)
def clean_up(self):
# start simulation cleanup
# delete this slave
print("goodbye....")
self._docean.request_delete(self._config['droplet_id'])
