Comment tester si la valeur int existe dans Python Enum sans utiliser try / catch?

Question

À l'aide de la classe Python Enum, existe-t-il un moyen de vérifier si un Enum contient une valeur int spécifique sans utiliser try / catch?

Avec la classe suivante:

 from enum import Enum

class Fruit(Enum):
    Apple = 4
    Orange = 5
    Pear = 6
 

Comment puis-je tester la valeur 6 (renvoyant true) ou la valeur 7 (renvoyant false)?

Answer 1

vous pouvez lister les valeurs comme ceci:

values = [item.value for item in Fruit]  # [4, 5, 6]

ou (probablement le meilleur): utiliser un set; l' in opérateur sera plus efficace:

values = set(item.value for item in Fruit)  # {4, 5, 6}

puis test avec

5 in values  # True
7 in values  # False

vous pouvez également ajouter ce que une méthode à votre classe:

class Fruit(Enum):
    Apple = 4
    Orange = 5
    Pear = 6

    @classmethod
    def has_value(cls, value):
        return any(value == item.value for item in cls)


print(Fruit.has_value(5))  # True
print(Fruit.has_value(7))  # False

si vous voulez tester pour les noms (et pas les valeurs) , Reda Maachi la réponse de couvre ce que je ferais.

Answer 2

Vous pouvez utiliser Enum.__members__ - un nom de dictionnaire dictionnaire mappé aux membres :

 In [12]: 'Apple' in Fruit.__members__
Out[12]: True

In [13]: 'Grape' in Fruit.__members__
Out[13]: False
 

Answer 3

Il suffit de vérifier si c'est en Enum. _value2member_map_

 In[15]: Fruit._value2member_map_
Out[15]: {4: <Fruit.Apple: 4>, 5: <Fruit.Orange: 5>, 6: <Fruit.Pear: 6>}

In[16]: 6 in Fruit._value2member_map_
Out[16]: True

In[17]: 7 in Fruit._value2member_map_
Out[17]: False
 

Answer 4

S'appuyant sur ce que Reda Maachi a commencé:

 6 in Fruit.__members__.values() 
 

renvoie True

 7 in Fruit.__members__.values()  
 

renvoie False