class LookupDictionary(dict): """ a dictionary which can lookup value by key, or keys by value """ def __init__(self, items=[]): """items can be a list of pair_lists or a dictionary""" dict.__init__(self, items) def get_keys_for_value(self, value, fail_value=None): """find the key(s) as a list given a value""" list_result = [item[0] for item in self.items() if item[1] == value] if len(list_result) > 0: return list_result return fail_value def get_first_key_for_value(self, value, fail_value=None): """return the first key of this dictionary given the value""" list_result = [item[0] for item in self.items() if item[1] == value] if len(list_result) > 0: return list_result[0] return fail_value def get_value(self, key, fail_value=None): """find the value given a key""" if key in self: return self[key] return fail_value class Enum(LookupDictionary): def __init__(self, initial_value=0, items=[]): """items can be a list of pair_lists or a dictionary""" LookupDictionary.__init__(self, items) self.value = initial_value def set_value(self, v): v_typename = typeof(v).__name__ if v_typename == 'str': if str in self: v = self[v] else: v = 0 else: self.value = v def get_enum_value(self): return self.value def get_enum_name(self): return self.__str__() def __str__(self): s = self.get_first_key_for_value(self.value, None) if s is None: s = "%#8.8x" % self.value return s def __repr__(self): return self.__str__()