Comment convertir int Enum en python?

À l'aide de la nouvelle Enum fonction (via backport enum34) avec python 2.7.6.

Donné la définition suivante, comment puis-je convertir un int correspondant à la valeur d'Enum?

from enum import Enum

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

Je sais que je peux fabriquer a la main une série de si-états pour faire la conversion, mais est-il facile pythonic moyen de convertir? En gros, je voudrais une fonction ConvertIntToFruit(int) renvoie une valeur d'enum.

Mon cas d'utilisation est que j'ai un fichier csv des dossiers où je suis en train de lire chaque enregistrement dans un objet. L'un des champs des fichiers de est un champ de type entier qui représente une énumération. Comme je suis le remplissage de l'objet que j'aimerais convertir en que le champ entier du fichier dans la valeur d'Enum dans l'objet.

InformationsquelleAutor User | 2014-05-30

98

Vous "appel" les Enum classe:
```
Fruit(5)
```
à son tour 5 en Fruit.Orange:
```
>>> from enum import Enum
>>> class Fruit(Enum):
...     Apple = 4
...     Orange = 5
...     Pear = 6
... 
>>> Fruit(5)
<Fruit.Orange: 5>
```
De la L'accès par programme aux membres de l'énumération et de leurs attributs section de la documentation:
Il est parfois utile d'accéder aux membres dans les énumérations
par programme (c'est à dire des situations où Color.red de ne pas le faire parce que l'
la couleur exacte n'est pas connue au programme-temps d'écriture). Enum permet
accès:
```
>>> Color(1)
<Color.red: 1>
>>> Color(3)
<Color.blue: 3>
```
Dans une note connexe: à la carte d'une valeur de chaîne contenant le nom d'un membre enum, l'utilisation d'abonnement:
```
>>> s = 'Apple'
>>> Fruit[s]
<Fruit.Apple: 4>
```
- Merci tout eu ce que j'ai été re-lecture de la documentation. N'était pas clair pour moi lors de ma première parcourir. Aussi, j'ai juste essayé à l'aide d'une chaîne et agréable de voir qu'il fonctionne lorsque la valeur de l'énumération est une chaîne de caractères en tant que bien et je suppose que n'importe quel objet de valeur.
- Très bon, si la conversion est en valeur. Est-il un analogue par son nom autrement que par une compréhension de liste à filtrer à travers l'énumération de la liste et de prendre le match sur les attributs de nom?
- faire suivre la documentation lien dans ma réponse, il est explicitement couvert. Utiliser point d'accès: Fruit['Orange'].
- Bingo, très impressionnant! J'ai cherché sur la page pour convert,coerce, et cast mais n'a pas touché quoi que ce soit. Ressemble à de la magie était access enum members by name, use item access. Perfecto, merci beaucoup, va nettoyer mon code avec cette syntaxe.
- Encore plus cool, c'est que je peux utiliser des trucs comme "foobar" in Fruit.__members__ de test pour l'adhésion et même arracher Fruit.get( 'foobar', Fruit.Orange ) pour obtenir que par défaut Enum saveur. Mon code est beaucoup plus simplifié en supprimant tous ces accessoires de recherche de échafaudages je me suis couchée plus tôt.
InformationsquelleAutor Martijn Pieters

Je pense que c'est en termes simples, est de convertir int valeur dans Enum en appelant EnumType(int_value), après que l'accès à la name de la Enum objet:

my_fruit_from_int = Fruit(5) #convert to int
fruit_name = my_fruit_from_int.name #get the name
print(fruit_name) #Orange will be printed here

Ou comme une fonction:

def convert_int_to_fruit(int_value):
    try:
        my_fruit_from_int = Fruit(int_value)
        return my_fruit_from_int.name
    except:
        return None

InformationsquelleAutor Ali Ezzat Odeh

Vous devez vous connecter pour publier un commentaire.