¿Cómo obtener todos los valores de la clase de enumeración de python?

Question 1

Estoy usando la biblioteca Enum4 para crear una clase de enumeración de la siguiente manera:

class Color(Enum):
    RED = 1
    BLUE = 2

quiero imprimir [1, 2] como una lista en alguna parte. ¿Cómo puedo conseguir esto?

Question 2

Puedes hacer lo siguiente:

[e.value for e in Color]

Question 3

Basado en la respuesta de @Jeff, refactorizado para usar un classmethod para que pueda reutilizar el mismo código para cualquiera de sus enumeraciones:

from enum import Enum

class ExtendedEnum(Enum):

    @classmethod
    def list(cls):
        return list(map(lambda c: c.value, cls))

class OperationType(ExtendedEnum):
    CREATE = 'CREATE'
    STATUS = 'STATUS'
    EXPAND = 'EXPAND'
    DELETE = 'DELETE'

print(OperationType.list())

Produce:

['CREATE', 'STATUS', 'EXPAND', 'DELETE']

Question 4

Puedes usar IntEnum:

from enum import IntEnum

class Color(IntEnum):
   RED = 1
   BLUE = 2


print(int(Color.RED))   # prints 1

Para obtener una lista de los enteros:

enum_list = list(map(int, Color))
print(enum_list) # prints [1, 2]

Question 5

Usar _member_names_ para un resultado rápido y fácil si son solo los nombres, es decir

Color._member_names_

Además, tienes _member_map_ que devuelve un diccionario ordenado de los elementos. Esta función devuelve un collections.OrderedDictasí que tienes Color._member_map_.items() y Color._member_map_.values() jugar con. P.ej

return list(map(lambda x: x.value, Color._member_map_.values()))

devolverá todos los valores válidos de Color

Question 6

Para usar Enum con cualquier tipo de valor, intente esto:
Actualizado con algunas mejoras… ¡Gracias @Jeff, por tu sugerencia!

from enum import Enum

class Color(Enum):
    RED = 1
    GREEN = 'GREEN'
    BLUE = ('blue', '#0000ff')

    @staticmethod
    def list():
        return list(map(lambda c: c.value, Color))

print(Color.list())

Como resultado:

[1, 'GREEN', ('blue', '#0000ff')]

Question 7

clase enum.Enum es una clase que resuelve todas sus necesidades de enumeración, por lo que solo necesita heredarla y agregar sus propios campos. Luego, a partir de ese momento, todo lo que necesita hacer es simplemente llamar a sus atributos: name & value:

from enum import Enum

class Letter(Enum):
   A = 1
   B = 2
   C = 3

print({i.name: i.value for i in Letter})
# prints {'A': 1, 'B': 2, 'C': 3}

Question 8

Usando un classmethod con __members__:

class RoleNames(str, Enum):
    AGENT = "agent"
    USER = "user"
    PRIMARY_USER = "primary_user"
    SUPER_USER = "super_user"
    
    @classmethod
    def list_roles(cls):
        role_names = [member.value for role, member in cls.__members__.items()]
        return role_names

>>> role_names = RoleNames.list_roles()
>>> print(role_names)

o si tienes varios Enum clases y desea abstraer la classmethod:

class BaseEnum(Enum):
    @classmethod
    def list_roles(cls):
        role_names = [member.value for role, member in cls.__members__.items()]
        return role_names


class RoleNames(str, BaseEnum):    
    AGENT = "agent"
    USER = "user"
    PRIMARY_USER = "primary_user"
    SUPER_USER = "super_user"
    

class PermissionNames(str, BaseEnum):
    READ = "updated_at"
    WRITE = "sort_by"
    READ_WRITE = "sort_order"

¿Ha sido útil esta solución?