How to implement a subscriptable class in Python (subscriptable class, not subscriptable object)?

Question

To implement a subscriptable object is easy, just implement __getitem__ in this object's class definition.
But now I want to implement a subscriptable class. For example, I want to implement this code:

class Fruit(object):     Apple = 0     Pear = 1     Banana = 2     #________________________________      #/ Some other definitions,         \     #\ make class 'Fruit' subscriptable. /     # --------------------------------      #        \   ^__^     #         \  (oo)\_______     #            (__)\       )\/\     #                ||----w |     #                ||     ||  print Fruit['Apple'], Fruit['Banana'] #Output: 0 2 

I know getattr can do the same thing, but I feel subscript accessing is more elegant.

Answer 1

Add something like this to your class:

class Fruit(object):      def __init__(self):          self.Fruits = {"Apple": 0, "Pear": 1, "Banana": 2}      def __getitem__(self, item):          return self.Fruits[item]