Support indexing and iterating Enum type objects

[JukkaL]

Assume this definition of Color:

from enum import Enum
class Color(Enum):
    red = 1
    green = 2
    blue = 3

Now this should be valid, but mypy complains about it:

Color['red']

The error message indicates that mypy thinks that the programmer is trying to construct a generic type, which is not the case here:

$ mypy t.py
t.py:8: error: Name 'red' is not defined

[johnthagen]

@JukkaL Another related Enum issue is that Mypy currently doesn't understand that Enum's can be iterated on.

import enum


@enum.unique
class Color(enum.Enum):
    red = 1
    green = 2
    blue = 3

for color in Color:
    print(color)

When run:

Color.red
Color.green
Color.blue

When mypy 0.4.4 is run on the file:

a.py:10: error: Iterable expected
a.py:10: error: "Color" has no attribute "__iter__"

[gvanrossum]

See also #2305.

[choucavalier]

Is someone working on this?

[gvanrossum]

Is someone working on this?

Not yet.

[JukkaL]

Increasing priority since this seems to be among the most common issues mypy users encounter.