def tostring(cls, val):
"Transforms val into string."
from Gaugi.utilities import get_attributes
for k, v in get_attributes(cls, getProtected = False):
if v==val:
return k
return None
def fromstring(cls, str_):
"Transforms string into enumeration."
from Gaugi.utilities import get_attributes
if not cls._ignoreCase:
return getattr(cls, str_, None)
else:
allowedValues = [attr for attr in get_attributes(cls) if not attr[0].startswith('_')]
try:
idx = [attr[0].upper() for attr in allowedValues].index(str_.upper().replace('-','_'))
except ValueError:
raise ValueError("%s is not in enumeration. Use one of the followings: %r" % (str_, allowedValues) )
return allowedValues[idx][1]
def retrieve(cls, val):
"""
Retrieve int value and check if it is a valid enumeration string or int on
this enumeration class.
"""
from Gaugi.utilities import get_attributes
allowedValues = [attr for attr in get_attributes(cls) if not attr[0].startswith('_')]
try:
# Convert integer string values to integer, if possible:
val = int(val)
except ValueError:
pass
if type(val) is str:
oldVal = val
val = cls.fromstring(val)
if val is None:
raise ValueError("String (%s) does not match any of the allowed values %r." % \
(oldVal, allowedValues))
else:
if not val in [attr[1] for attr in allowedValues]:
raise ValueError(("Attempted to retrieve val benchmark "
"with a enumeration value which is not allowed. Use one of the followings: "
"%r") % allowedValues)
return val
def intList(cls): from operator import itemgetter from Gaugi.utilities import get_attributes return [v[1] for v in sorted(get_attributes( cls, getProtected = False), key=itemgetter(1))]