Constantes de classe en python

Question

En python, je veux qu'une classe ait des "constantes" (pratiquement des variables) qui seront communes dans toutes les sous-classes. Y a-t-il un moyen de le faire avec une syntaxe amicale ? En ce moment j'utilise :

class Animal:
    SIZES=["Huge","Big","Medium","Small"]

class Horse(Animal):
    def printSize(self):
        print(Animal.SIZES[1])

et je me demande s'il y a une meilleure façon de le faire ou une façon de le faire sans avoir à écrire "Animal" avant les tailles. Cheval hérité de Animal.

Answer 1

Vous pouvez accéder à SIZES au moyen de self.SIZES (dans une méthode d'instance) ou cls.SIZES (dans une méthode de classe).

Dans tous les cas, vous devrez être explicite sur l'endroit où trouver SIZES. Une alternative est de mettre SIZES dans le module contenant les classes, mais ensuite vous devez définir toutes les classes dans un seul module.

Answer 2

class Animal:
    HUGE = "Huge"
    BIG = "Big"

class Horse:
    def printSize(self):
        print(Animal.HUGE)

Answer 3

En développant la réponse de betabandido, vous pouvez écrire une fonction pour injecter les attributs en tant que constantes dans le module :

def module_register_class_constants(klass, attr_prefix):
    globals().update(
        (name, getattr(klass, name)) for name in dir(klass) if name.startswith(attr_prefix)
    )

class Animal(object):
    SIZE_HUGE = "Huge"
    SIZE_BIG = "Big"

module_register_class_constants(Animal, "SIZE_")

class Horse(Animal):
    def printSize(self):
        print SIZE_BIG