diff options
| author | Martin Peres <martin.peres@linux.intel.com> | 2017-03-25 23:16:51 +0200 |
|---|---|---|
| committer | Martin Peres <martin.peres@linux.intel.com> | 2017-10-04 17:48:28 +0300 |
| commit | 8ac4956f3b1ba0d2ade166ee10d6104eca880517 (patch) | |
| tree | 3713e3184bd9c84c82f1ec9840dbe25dfa2d59bb | |
| parent | 62285fe41c3d52a7076a0266645faa6bb6b2464f (diff) | |
smartezbench: allow registering a callback for logs
| -rw-r--r-- | python-modules/ezbench/smartezbench.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/python-modules/ezbench/smartezbench.py b/python-modules/ezbench/smartezbench.py index 03a8842..f2345c6 100644 --- a/python-modules/ezbench/smartezbench.py +++ b/python-modules/ezbench/smartezbench.py @@ -213,12 +213,13 @@ class SmartEzbenchAttributes(Enum): class SmartEzbench: def __init__(self, ezbench_dir, report_name, readonly = False, - hook_binary_path = None): + hook_binary_path = None, logs_callback = None): self.readonly = readonly self.ezbench_dir = ezbench_dir self.report_name = report_name self.log_folder = ezbench_dir + '/logs/' + report_name self.hook_binary_path = hook_binary_path + self.logs_callback = logs_callback self.smart_ezbench_state = self.log_folder + "/smartezbench.state" self.smart_ezbench_lock = self.log_folder + "/smartezbench.lock" self.smart_ezbench_log = self.log_folder + "/smartezbench.log" @@ -258,6 +259,8 @@ class SmartEzbench: log_msg = "{time}: ({error}) {msg}\n".format(time=time, error=error.name, msg=msg) if error.value <= self.min_criticality.value: print(log_msg, end="") + if self.logs_callback is not None: + self.logs_callback(self, log_msg) if not self.readonly: self.log_file.write(log_msg) self.log_file.flush() |