shutil.copytree(cdes_dir, cdes_scen_dir)
# Directories are the same
except shutil.Error as e:
logging.error(
"CDES_Loader(): Main(): Directory not copied. Error: %s" % e)
exit()
# Any error saying that the directory doesn't exist
except OSError as e:
logging.error(
"CDES_Loader(): Main(): Directory not copied. Error: %s" % e)
exit()
###Get node states
sr = SessionReader(session_number)
# First check to make sure the session is in a running state
state = sr.get_session_state()
if state == None:
logging.error(
"CDES_Loader(): Main(): Session " + str(session_number) +
" does not exist, make sure core-daemon is running. \n You can start it by running /etc/init.d/core-daemon start"
)
exit()
while "4 RUNTIME_STATE" not in state:
logging.warning(
"CDES_Loader(): Main(): Session " + str(session_number) +
" Is not yet in the running state... waiting and then trying again"
)
if state == None:
logging.error(
"CDES_Loader(): Main(): Session " + str(session_number) +
" does not exist, make sure core-daemon is running. \n You can start it by running /etc/init.d/core-daemon start"
def cdes_run(self,
session_number=None,
monitor_cmd=None,
conditional_conns=None):
logging.debug("Controller(): cdes_run(): Instantiated")
if monitor_cmd == None:
logging.error(
"Controller(): cdes_run(): No monitor command was passed in, quitting"
)
exit()
cmd = os.path.abspath(monitor_cmd)
if session_number == None:
logging.error(
"Controller(): cdes_run(): No session number was passed in, quitting"
)
exit()
sr = SessionReader(session_number)
if conditional_conns == None:
conditional_conns = sr.relevant_session_to_JSON()
imonitor_queues = []
omonitor_queues = []
otrigger_queues = []
oswapper_queues = []
monitor_processes = []
trigger_processes = []
swapper_processes = []
for cc_dec_node in conditional_conns.keys():
logging.debug("Controller(): cdes_run(): In loop")
# For knowning when to quit
imonitor_queue = multiprocessing.Queue()
omonitor_queue = multiprocessing.Queue()
otrigger_queue = multiprocessing.Queue()
oswapper_queue = multiprocessing.Queue()
imonitor_queues.append(imonitor_queue)
omonitor_queues.append(omonitor_queue)
otrigger_queues.append(otrigger_queue)
oswapper_queues.append(oswapper_queue)
m = Monitor("monitor", imonitor_queue, omonitor_queue, cmd)
mp = multiprocessing.Process(target=m.run_monitor)
mp.start()
monitor_processes.append(mp)
#If the user defined a custom Trigger, load it now
#get cc_dec name
cc_dec_name = conditional_conns[cc_dec_node]["name"]
filepath = os.path.realpath(__file__)
file_dir = os.path.dirname(filepath)
custom_file_path = os.path.join(file_dir, "Trigger", cc_dec_name,
"MyTrigger.py")
if os.path.exists(custom_file_path):
DynLoadedClass = imp.load_source('MyTrigger', custom_file_path)
tp = DynLoadedClass.MyTrigger("trigger", omonitor_queue,
otrigger_queue,
conditional_conns, cc_dec_node)
else:
tp = TimerTrigger("trigger", omonitor_queue, otrigger_queue,
conditional_conns, cc_dec_node)
tpm = multiprocessing.Process(target=tp.process_data)
tpm.start()
trigger_processes.append(tpm)
short_session_number = self.short_session_id(session_number)
sw = Swapper("swapper", otrigger_queue, oswapper_queue,
conditional_conns, session_number,
short_session_number, cc_dec_node)
swp = multiprocessing.Process(target=sw.update_connection)
swp.start()
swapper_processes.append(swp)
# Keep looping until the scenario is no longer in a Run state
state = sr.get_session_state()
if state == None:
logging.error("Controller(): cdes_run(): Session " +
str(session_number) + " no longer exists, exitting")
exit()
while "4 RUNTIME_STATE" in state:
logging.debug("Controller(): cdes_run(): Session " +
str(session_number) +
" session Still running, continuing operations")
if state == None:
logging.error("Controller(): cdes_run(): Session " +
str(session_number) +
" no longer exists, exitting")
break
time.sleep(5)
state = sr.get_session_state()
# Logic to terminate all processes goes here
for cc in range(0, len(monitor_processes)):
logging.debug(
"Controller(): cdes_run(): Cleaning up and exiting " +
str(session_number))
imonitor_queues[cc].put("exit")
logging.debug(
"Controller(): cdes_run(): Waiting for Monitor process to fininsh to gracefully exit"
)
monitor_processes[cc].join()
logging.debug(
"Controller(): cdes_run(): Done. Terminating Monitor.")
monitor_processes[cc].terminate()
logging.debug(
"Controller(): cdes_run(): Done. Terminating Trigger.")
trigger_processes[cc].terminate()
logging.debug(
"Controller(): cdes_run(): Done. Terminating Swapper.")
swapper_processes[cc].terminate()
logging.debug("Controller(): cdes_run(): Done.")