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()