def perform_task(
scenario_id, tasktype_id, worker_nr, broker_logging_handler=None):
"""
execute specific task
scenario_id = id of scenario
tasktype_id = id of tasktype (120,130,132)
worker_nr = number of worker (1,2,3,4,5,6,7,8). Used for temp
directory and sobek project.
broker_logging_handler = sends loggings to broker
"""
#logging handler
if broker_logging_handler is not None:
log.addHandler(broker_logging_handler)
else:
log.warning("Broker logging handler does not set.")
#settings.py:
hisssm_root = settings.HISSSM_ROOT
sobek_program_root = settings.SOBEK_PROGRAM_ROOT # e: or c:
#sobek_project_root = settings.SOBEK_PROJECT_ROOT
tmp_root = settings.TMP_ROOT
tmp_directory = os.path.join(tmp_root, str(worker_nr))
sobek_project_directory = os.path.join(
'lzfl_%03d.lit' % worker_nr)
try:
success_code = False
error_message = ""
remarks = ''
if tasktype_id == TASK_COMPUTE_SOBEK_MODEL_120:
log.debug("execute TASK_COMPUTE_SOBEK_MODEL_120")
from flooding_worker.tasks import openbreach
openbreach.set_broker_logging_handler(broker_logging_handler)
remarks = ('openbreach-' + openbreach.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = openbreach.compute_sobek_model(scenario_id,
tmp_directory)
elif tasktype_id == TASK_PERFORM_SOBEK_SIMULATION_130:
log.debug("execute TASK_PERFORM_SOBEK_SIMULATION_130")
from flooding_worker.tasks import spawn
spawn.set_broker_logging_handler(broker_logging_handler)
remarks = ('spawn-' + spawn.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = spawn.perform_sobek_simulation(
scenario_id,
sobek_project_directory,
sobek_program_root)
elif tasktype_id == TASK_SOBEK_PNG_GENERATION_150:
log.debug("execute TASK_SOBEK_PNG_GENERATION_150")
from flooding_worker.tasks import png_generation
png_generation.set_broker_logging_handler(broker_logging_handler)
remarks = ('png_generation-' + png_generation.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = png_generation.sobek(scenario_id, tmp_directory)
elif tasktype_id == TASK_COMPUTE_RISE_SPEED_132:
log.debug("execute TASK_COMPUTE_RISE_SPEED_132")
from flooding_worker.tasks import calculaterisespeed_132
calculaterisespeed_132.set_broker_logging_handler(
broker_logging_handler)
remarks = ('calculaterisespeed_132-' +
calculaterisespeed_132.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = calculaterisespeed_132.perform_calculation(
scenario_id, tmp_directory)
elif tasktype_id == TASK_COMPUTE_MORTALITY_GRID_134:
log.debug("execute TASK_COMPUTE_MORTALITY_GRID_134")
from flooding_worker.tasks import calculatemortalitygrid_134
calculatemortalitygrid_134.set_broker_logging_handler(
broker_logging_handler)
remarks = ('calculatemortalitygrid_134-' +
calculatemortalitygrid_134.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = calculatemortalitygrid_134.perform_calculation(
scenario_id, tmp_directory)
elif tasktype_id == TASK_SOBEK_PRESENTATION_GENERATION_155:
log.debug("execute TASK_SOBEK_PRESENTATION_GENERATION_155")
from flooding_worker.tasks import presentationlayer_generation
presentationlayer_generation.set_broker_logging_handler(
broker_logging_handler)
remarks = ('presentationlayer_generation-' +
presentationlayer_generation.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = (
presentationlayer_generation.perform_presentation_generation(
scenario_id, tasktype_id))
elif tasktype_id == TASK_HISSSM_SIMULATION_160:
log.debug("execute TASK_HISSSM_SIMULATION_160")
from flooding_worker.tasks import hisssm_160
hisssm_160.set_broker_logging_handler(broker_logging_handler)
remarks = ('hisssm_160-' + hisssm_160.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = hisssm_160.perform_HISSSM_calculation(scenario_id,
hisssm_root)
elif tasktype_id == TASK_SOBEK_EMBANKMENT_DAMAGE_162:
log.debug("execute TASK_SOBEK_EMBANKMENT_DAMAGE_162")
from flooding_worker.tasks import kadeschade_module
kadeschade_module.set_broker_logging_handler(
broker_logging_handler)
remarks = ('kadeschade_module-' + kadeschade_module.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code, error_message = kadeschade_module.calc_damage(
scenario_id)
elif tasktype_id == TASK_HISSSM_PNG_GENERATION_180:
log.debug("execute TASK_HISSSM_PNG_GENERATION_180")
from flooding_worker.tasks import png_generation
png_generation.set_broker_logging_handler(broker_logging_handler)
remarks = ('png_generation-' + png_generation.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = png_generation.his_ssm(scenario_id, tmp_directory)
elif tasktype_id == TASK_HISSSM_PRESENTATION_GENERATION_185:
log.debug("execute TASK_HISSSM_PRESENTATION_GENERATION_185")
from flooding_worker.tasks import presentationlayer_generation
presentationlayer_generation.set_broker_logging_handler(
broker_logging_handler)
remarks = ('presentationlayer_generation-' +
presentationlayer_generation.__revision__ +
' uitvoerder: %02d/' % worker_nr)
success_code = (
presentationlayer_generation.perform_presentation_generation(
scenario_id, tasktype_id))
elif tasktype_id == TASK_CALCULATE_STATISTICS:
from flooding_worker.tasks import calculate_scenario_statistics
calculate_scenario_statistics.calculate_statistics(scenario_id)
success_code = True # In case of problems, an exception is raised
else:
log.warning("selected a '%d' task but don't know what it is" %
tasktype_id)
remarks = 'unknown task'
return success_code, remarks, error_message
except Exception, e:
from sys import exc_info
from traceback import format_tb
(this_exctype, this_value, this_traceback) = exc_info()
log.warning(''.join(['traceback: \n'] + format_tb(this_traceback)))
log.error("while executing task %s: '%s(%s)'" %
(tasktype_id, type(e), str(e)))
return False, remarks, '%s(%s)' % (type(e), str(e))
