Class decorator for convenient creation of class attribute-based enumeration constants.

enum_class.py

def enum_class(cls):
    """Decorates a class to set it's attributes to values of their
    literal names. This is cool to use to make an 'enum' class and
    have PyCharm also infer it's members.

    Use it like so:
    >>> @enum_class
    ... class KindOfFruit:
    ...     Apple, Orange, Banana, Passionfruit = range(4)
    ...
    >>> orange_kind = KindOfFruit.Orange
    >>> print orange_kind
    Orange
    """
    for attr in dir(cls):
        if not attr.startswith('__'):
            setattr(cls, attr, attr)
    return cls