web-dev-qa-db-fra.com

Quelle est la pratique courante pour les enums en Python?

Duplicate possible:
Comment puis-je représenter un "enum" en Python?

Quelle est la pratique courante pour les enums en Python? C'est à dire. comment sont-ils répliqués en Python?

public enum Materials
{
    Shaded,
    Shiny,
    Transparent,
    Matte
}
pythonenums
136
Joan Venge
class Materials:
    Shaded, Shiny, Transparent, Matte = range(4)

>>> print Materials.Matte
3
363
Van Gale

J'ai vu ce modèle plusieurs fois:

>>> class Enumeration(object):
        def __init__(self, names):  # or *names, with no .split()
            for number, name in enumerate(names.split()):
                setattr(self, name, number)

>>> foo = Enumeration("bar baz quux")
>>> foo.quux
2

Vous pouvez également utiliser les membres du groupe, mais vous devrez fournir votre propre numérotation:

>>> class Foo(object):
        bar  = 0
        baz  = 1
        quux = 2

>>> Foo.quux
2

Si vous recherchez quelque chose de plus robuste (valeurs éparses, exception spécifique à une énumération, etc.), essayez cette recette .

19
Ben Blank

Je ne comprends pas pourquoi les enums ne sont pas supportés nativement par Python. La meilleure façon que j'ai trouvée de les émuler est de surcharger _ str _ et _ eq _ pour pouvoir les comparer et lorsque vous utilisez print (), vous obtenez la chaîne au lieu de la valeur numérique.

class enumSeason():
    Spring = 0
    Summer = 1
    Fall = 2
    Winter = 3
    def __init__(self, Type):
        self.value = Type
    def __str__(self):
        if self.value == enumSeason.Spring:
            return 'Spring'
        if self.value == enumSeason.Summer:
            return 'Summer'
        if self.value == enumSeason.Fall:
            return 'Fall'
        if self.value == enumSeason.Winter:
            return 'Winter'
    def __eq__(self,y):
       return self.value==y.value

Usage:

>>> s = enumSeason(enumSeason.Spring)

>>> print(s)

Spring
9
Spell

Vous pourriez probablement utiliser une structure d'héritage, bien que plus je joue avec cela, plus je me sens sale.

class AnimalEnum:
  @classmethod
  def verify(cls, other):
    return issubclass(other.__class__, cls)


class Dog(AnimalEnum):
  pass

def do_something(thing_that_should_be_an_enum):
  if not AnimalEnum.verify(thing_that_should_be_an_enum):
    raise OhGodWhy
6
Trey Stout