sascha/workload-automation
1
0
Fork 0
mirror of https://github.com/ARM-software/workload-automation.git synced 2025-09-02 03:12:34 +01:00
Code Issues Releases Wiki Activity

util/exec_control: add once_per_attribute_value

Browse Source 
Add a decorator to run a method once for all instances that share the
value of the specified attribute.
This commit is contained in:
Sergei Trofimov
2020-02-07 16:30:04 +00:00
committed by Marc Bonnici
parent 51ffd60c06
commit 0f2de5f951
2 changed files with 65 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
wa/utils/exec_control.py
View File

@@ -105,6 +105,30 @@ def once_per_class(method):
return wrapper
def once_per_attribute_value(attr_name):
"""
The specified method will be invoked once for all instances that share the
same value for the specified attribute (sameness is established by comparing
repr() of the values).
"""
def wrapped_once_per_attribute_value(method):
def wrapper(*args, **kwargs):
if __active_environment is None:
activate_environment('default')
attr_value = getattr(args[0], attr_name)
func_id = repr(method.__name__) + repr(args[0].__class__) + repr(attr_value)
if func_id in __environments[__active_environment]:
return
else:
__environments[__active_environment].append(func_id)
return method(*args, **kwargs)
return wrapper
return wrapped_once_per_attribute_value
def once(method):
"""
The specified method will be invoked only once within the