bitcointools/enumeration.py

   1 #
   2 # enum-like type
   3 # From the Python Cookbook, downloaded from http://code.activestate.com/recipes/67107/
   4 #
   5 import types, string, exceptions
   6
   7 class EnumException(exceptions.Exception):
   8     pass
   9
  10 class Enumeration:
  11     def __init__(self, name, enumList):
  12         self.__doc__ = name
  13         lookup = { }
  14         reverseLookup = { }
  15         i = 0
  16         uniqueNames = [ ]
  17         uniqueValues = [ ]
  18         for x in enumList:
  19             if type(x) == types.TupleType:
  20                 x, i = x
  21             if type(x) != types.StringType:
  22                 raise EnumException, "enum name is not a string: " + x
  23             if type(i) != types.IntType:
  24                 raise EnumException, "enum value is not an integer: " + i
  25             if x in uniqueNames:
  26                 raise EnumException, "enum name is not unique: " + x
  27             if i in uniqueValues:
  28                 raise EnumException, "enum value is not unique for " + x
  29             uniqueNames.append(x)
  30             uniqueValues.append(i)
  31             lookup[x] = i
  32             reverseLookup[i] = x
  33             i = i + 1
  34         self.lookup = lookup
  35         self.reverseLookup = reverseLookup
  36     def __getattr__(self, attr):
  37         if not self.lookup.has_key(attr):
  38             raise AttributeError
  39         return self.lookup[attr]
  40     def whatis(self, value):
  41         return self.reverseLookup[value]