enum.py

def enum(name, attrs):
    """
    >>> Fruit = enum('Fruit', 'apple banana cranberry')
    >>> Fruit.apple  #doctest: +ELLIPSIS
    <object object at 0x...>
    >>> Fruit.apple == Fruit.banana
    False
    >>> Animal = enum('Animal', 'cat dog elephant')
    >>> Animal.cat  #doctest: +ELLIPSIS
    <object object at 0x...>
    >>> Animal.dog == Animal.elephant
    False
    >>> Animal.cat == Fruit.apple
    False
    """
    return type(name, (object, ), {a: object() for a in attrs.split(' ')})


if __name__ == '__main__':
    import doctest
    doctest.testmod()