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

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

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]

Seems to work by changing the metaclass. For Python 2:

class GetAttr(type):
    def __getitem__(cls, x):
        return getattr(cls, x)

class Fruit(object):
    __metaclass__ = GetAttr

    Apple = 0
    Pear = 1
    Banana = 2

print Fruit[Apple], Fruit[Banana]
# output: 0 2

---

On Python 3, you should use Enum directly:

import enum

class Fruit(enum.Enum):
    Apple = 0
    Pear = 1
    Banana = 2

print(Fruit[Apple], Fruit[Banana])
# Output: Fruit.Apple, Fruit.Banana
print(Fruit[Apple].value, Fruit[Banana].value)
# Output: 0 2

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

Expanding on @LuisKleinworts answer, if you want to do this for all class attributes:

fruits_dict = {apple:0, banana:1}

class Fruits(object):
    def __init__(self, args):
        for k in args:
            setattr(self, k, args[k])
            
    def __getitem__(self, item):
        return getattr(self, item)

fruits = Fruits(fruits_dict)
print(fruits.apple)
print(fruits[apple])