class Session(object):
def __init__(self, uplink, config, session_id, hydra_url):
'''
Constructor
'''
self.uplink = uplink
self.config = config
self.session_id = session_id
self.hydra_url = hydra_url
self.setup = Setup(self)
if self.config.has_option('general', 'fake'):
self.fake = (self.config.get('general', 'fake') == "yes")
else:
self.fake = False
def log_format(self, message):
return self.uplink.log_format("[" + self.session_id + "] " + message)
def prepare(self):
logging.info(self.setup.log_format(("preparing setup")))
if not self.fake:
self.setup.load()
self.setup.prepare_base()
def add_node(self, node_id, node_name, ip_address, netmask):
try:
logging.info(
self.setup.log_format(("add node " + str(node_id) + " '" +
str(node_name) + "'")))
if not self.fake:
self.setup.add_node(node_id, node_name, (ip_address, netmask))
except ValueError:
pass
def remove_node(self, node_id):
logging.info(self.setup.log_format(("remove node " + str(node_id))))
if not self.fake:
self.setup.remove_node(node_id)
def action(self, data):
try:
logging.info(self.setup.log_format(("call action: " + data)))
if not self.fake:
return self.setup.action(data)
except ValueError:
pass
def run(self):
""" run the nodes """
logging.info(self.setup.log_format(("run all the nodes")))
if not self.fake:
self.setup.startup()
def stop(self):
""" stop the nodes """
logging.info(self.setup.log_format(("stop all the nodes")))
if not self.fake:
self.setup.shutdown()
def cleanup(self):
""" cleanup the setup """
logging.info(self.setup.log_format(("cleaning up")))
""" stop all nodes """
if not self.fake:
self.setup.shutdown()
""" delete the setup folder """
if not self.fake:
self.setup.cleanup()
class Session(object):
def __init__(self, uplink, config, session_id, hydra_url):
'''
Constructor
'''
self.uplink = uplink
self.config = config
self.session_id = session_id
self.hydra_url = hydra_url
self.setup = Setup(self)
if self.config.has_option('general', 'fake'):
self.fake = (self.config.get('general', 'fake') == "yes")
else:
self.fake = False
def log_format(self, message):
return self.uplink.log_format("[" + self.session_id + "] " + message)
def prepare(self):
logging.info(self.setup.log_format(("preparing setup")))
if not self.fake:
self.setup.load()
self.setup.prepare_base()
def add_node(self, node_id, node_name, ip_address, netmask):
try:
logging.info(self.setup.log_format(("add node " + str(node_id) + " '" + str(node_name) + "'")))
if not self.fake:
self.setup.add_node(node_id, node_name, (ip_address, netmask))
except ValueError:
pass
def remove_node(self, node_id):
logging.info(self.setup.log_format(("remove node " + str(node_id))))
if not self.fake:
self.setup.remove_node(node_id)
def action(self, data):
try:
logging.info(self.setup.log_format(("call action: " + data)))
if not self.fake:
return self.setup.action(data)
except ValueError:
pass
def run(self):
""" run the nodes """
logging.info(self.setup.log_format(("run all the nodes")))
if not self.fake:
self.setup.startup()
def stop(self):
""" stop the nodes """
logging.info(self.setup.log_format(("stop all the nodes")))
if not self.fake:
self.setup.shutdown()
def cleanup(self):
""" cleanup the setup """
logging.info(self.setup.log_format(("cleaning up")))
""" stop all nodes """
if not self.fake:
self.setup.shutdown()
""" delete the setup folder """
if not self.fake:
self.setup.cleanup()
