mirror of
https://github.com/ARM-software/workload-automation.git
synced 2024-10-06 10:51:13 +01:00
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
from wa.framework import pluginloader
|
||
|
|
from wa.framework.exception import ConfigError
|
||
|
|
from wa.framework.instrumentation import is_installed
|
||
|
|
from wa.framework.plugin import Plugin
|
||
|
|
from wa.utils.log import log_error, indent, dedent
|
||
|
|
|
||
|
|
|
||
|
|
class ResultProcessor(Plugin):
|
||
|
|
|
||
|
|
kind = 'result_processor'
|
||
|
|
requires = []
|
||
|
|
|
||
|
|
def validate(self):
|
||
|
|
super(ResultProcessor, self).validate()
|
||
|
|
for instrument in self.requires:
|
||
|
|
if not is_installed(instrument):
|
||
|
|
msg = 'Instrument "{}" is required by {}, but is not installed.'
|
||
|
|
raise ConfigError(msg.format(instrument, self.name))
|
||
|
|
|
||
|
|
def initialize(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def finalize(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class ProcessorManager(object):
|
||
|
|
|
||
|
|
def __init__(self, loader=pluginloader):
|
||
|
|
self.loader = loader
|
||
|
|
self.logger = logging.getLogger('processor')
|
||
|
|
self.processors = []
|
||
|
|
|
||
|
|
def install(self, processor):
|
||
|
|
if not isinstance(processor, ResultProcessor):
|
||
|
|
processor = self.loader.get_result_processor(processor)
|
||
|
|
self.logger.debug('Installing {}'.format(processor.name))
|
||
|
|
self.processors.append(processor)
|
||
|
|
|
||
|
|
def validate(self):
|
||
|
|
for proc in self.processors:
|
||
|
|
proc.validate()
|
||
|
|
|
||
|
|
def initialize(self):
|
||
|
|
for proc in self.processors:
|
||
|
|
proc.initialize()
|
||
|
|
|
||
|
|
def finalize(self):
|
||
|
|
for proc in self.processors:
|
||
|
|
proc.finalize()
|
||
|
|
|
||
|
|
def process_job_output(self, context):
|
||
|
|
self.do_for_each_proc('process_job_output', 'processing using "{}"',
|
||
|
|
context.job_output, context.target_info,
|
||
|
|
context.run_output)
|
||
|
|
|
||
|
|
def export_job_output(self, context):
|
||
|
|
self.do_for_each_proc('export_job_output', 'Exporting using "{}"',
|
||
|
|
context.job_output, context.target_info,
|
||
|
|
context.run_output)
|
||
|
|
|
||
|
|
def process_run_output(self, context):
|
||
|
|
self.do_for_each_proc('process_run_output', 'Processing using "{}"',
|
||
|
|
context.run_output, context.target_info)
|
||
|
|
|
||
|
|
def export_run_output(self, context):
|
||
|
|
self.do_for_each_proc('export_run_output', 'Exporting using "{}"',
|
||
|
|
context.run_output, context.target_info)
|
||
|
|
|
||
|
|
def do_for_each_proc(self, method_name, message, *args):
|
||
|
|
try:
|
||
|
|
indent()
|
||
|
|
for proc in self.processors:
|
||
|
|
proc_func = getattr(proc, method_name, None)
|
||
|
|
if proc_func is None:
|
||
|
|
continue
|
||
|
|
try:
|
||
|
|
self.logger.info(message.format(proc.name))
|
||
|
|
proc_func(*args)
|
||
|
|
except Exception as e:
|
||
|
|
if isinstance(e, KeyboardInterrupt):
|
||
|
|
raise
|
||
|
|
log_error(e, self.logger)
|
||
|
|
finally:
|
||
|
|
dedent()
|