sascha/workload-automation
1
0
Fork 0
mirror of https://github.com/ARM-software/workload-automation.git synced 2025-02-21 20:38:57 +00:00
Code Issues Releases Wiki Activity

framework/execution: add context to job signals

Browse Source 
Pass the execution context to job execution signal callbacks.
This commit is contained in:
Sergei Trofimov 2018-01-25 08:40:44 +00:00 committed by Marc Bonnici
parent a2c113e0f9
commit dae4d9a934
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
wa/framework/execution.py
View File

@ -375,7 +375,7 @@ class Runner(object):
while self.context.job_queue: while self.context.job_queue:
try: try:
with signal.wrap('JOB_EXECUTION', self): with signal.wrap('JOB_EXECUTION', self, self.context):
self.run_next_job(self.context) self.run_next_job(self.context)
except KeyboardInterrupt: except KeyboardInterrupt:
self.context.skip_remaining_jobs() self.context.skip_remaining_jobs()
@ -446,16 +446,16 @@ class Runner(object):
job.set_status(Status.RUNNING) job.set_status(Status.RUNNING)
self.send(signal.JOB_STARTED) self.send(signal.JOB_STARTED)
with signal.wrap('JOB_TARGET_CONFIG', self): with signal.wrap('JOB_TARGET_CONFIG', self, context):
job.configure_target(context) job.configure_target(context)
with signal.wrap('JOB_SETUP', self): with signal.wrap('JOB_SETUP', self, context):
job.setup(context) job.setup(context)
try: try:
try: try:
with signal.wrap('JOB_EXECUTION', self): with signal.wrap('JOB_EXECUTION', self, context):
job.run(context) job.run(context)
except Exception as e: except Exception as e:
job.set_status(Status.FAILED) job.set_status(Status.FAILED)
@ -465,7 +465,7 @@ class Runner(object):
raise e raise e
finally: finally:
try: try:
with signal.wrap('JOB_OUTPUT_PROCESSED', self): with signal.wrap('JOB_OUTPUT_PROCESSED', self, context):
job.process_output(context) job.process_output(context)
self.pm.process_job_output(context) self.pm.process_job_output(context)
self.pm.export_job_output(context) self.pm.export_job_output(context)