classification
Title: add __iter__ to enum.Flag members
Type: enhancement Stage:
Components: Versions:
process
Status: open Resolution:
Dependencies: Superseder:
Assigned To: Nosy List: Guy Gangemi, ethan.furman
Priority: normal Keywords:

Created on 2017-12-05 03:44 by Guy Gangemi, last changed 2017-12-05 05:05 by berker.peksag.

Messages (1)
msg307629 - (view) Author: Guy Gangemi (Guy Gangemi) Date: 2017-12-05 03:44
I'm proposing to extend enum.Flag member functionality so it is iterable in a manner similar to enum.Flag subclasses.

from enum import Flag, auto


class FlagIter(Flag):
    def __iter__(self):
        for memeber in self._member_map_.values():
            if member in self:
                yield member


class Colour(FlagIter):
    RED = auto()
    GREEN = auto()
    BLUE = auto()
    YELLOW = RED | GREEN

cyan = Colour.GREEN | Colour.Blue

print(*Colour)  # Colour.RED Colour.GREEN Colour.BLUE Colour.YELLOW

# Without the enhancement, 'not iterable' is thrown for these
print(*cyan)  # Colour.GREEN Colour.BLUE
print(*Colour.YELLOW)  # Colour.RED Colour.GREEN Colour.YELLOW
print(*~Colour.RED)  # Colour.GREEN Colour.BLUE
History
Date User Action Args
2017-12-05 05:05:41berker.peksagsetnosy: + ethan.furman
2017-12-05 03:44:12Guy Gangemicreate