183

In python, I want a class to have some "constants" (practically, variables) which will be common in all subclasses. Is there a way to do it with friendly syntax? Right now I use:

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

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

and I'm wondering if there is a better way to do it or a way to do it without then having to write "Animal." before the sizes. Horse inherits from Animal.

janw
  • 8,758
  • 11
  • 40
  • 62
Idan
  • 5,365
  • 5
  • 24
  • 28
  • 29
    Not a full answer but If `SIZES` never changes, definitely use a tuple instead of a list. Like so `("Huge","Big","Medium","Small")` – jamylak May 20 '12 at 09:53
  • 3
    Looks like you're really after an enum here – Eric May 20 '12 at 11:39
  • @jamylak your comment seems to be popular, many people seem to consider it valueable. Could you explain the logic behind it? – dmigo Apr 29 '21 at 13:11
  • 3
    @dmigo Just that the question was asking about constants and a list can be mutated, so it's better to use a tuple which is immutable. A constant should not change – jamylak Apr 29 '21 at 13:46

4 Answers4

191

Since Horse is a subclass of Animal, you can just change

print(Animal.SIZES[1])

with

print(self.SIZES[1])

Still, you need to remember that SIZES[1] means "big", so probably you could improve your code by doing something like:

class Animal:
    SIZE_HUGE="Huge"
    SIZE_BIG="Big"
    SIZE_MEDIUM="Medium"
    SIZE_SMALL="Small"

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

Alternatively, you could create intermediate classes: HugeAnimal, BigAnimal, and so on. That would be especially helpful if each animal class will contain different logic.

pmaruszczyk
  • 2,157
  • 2
  • 24
  • 49
betabandido
  • 18,946
  • 11
  • 62
  • 76
27

You can get to SIZES by means of self.SIZES (in an instance method) or cls.SIZES (in a class method).

In any case, you will have to be explicit about where to find SIZES. An alternative is to put SIZES in the module containing the classes, but then you need to define all classes in a single module.

Fred Foo
  • 355,277
  • 75
  • 744
  • 836
20
class Animal:
    HUGE = "Huge"
    BIG = "Big"

class Horse:
    def printSize(self):
        print(Animal.HUGE)
amit kumar
  • 20,438
  • 23
  • 90
  • 126
  • Also you can access via self, ex. "self.HUGE" or "self.BIG". This would only work internally within the class. – cevaris Jan 08 '14 at 17:17
6

Expanding on betabandido's answer, you could write a function to inject the attributes as constants into the 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
Matthew Trevor
  • 14,354
  • 6
  • 37
  • 50