Python overloading multiple getitems / index requests

Question

I have a Grid class which I want to access using myGrid[1][2]. I know I can overload the first set of square brackets with the __getitem__()

Answer 1

This question is quite old, but I'll add my answer anyway for newcomers.
I ran into this need myself, and here's a solution that worked for me:

class Grid:

    def __init__(self):
        self.matrix = [[0,1,2],[3,4,5],[6,7,8]]
        self.second_access = False

    def __getitem__(self, key):
        if not self.second_access:
            self.first_key = key
            self.second_access = True
            return self
        else:
            self.second_access = False
            return self.matrix[self.first_key][key]


g = Grid()
print(g[1][2])  # 5
print(g[0][1])  # 1
print(g[2][0])  # 6

Notice that this will not work for single access!
So, for example, if you want to use something of the form g[0] to get [0,1,2] it will not work, and instead you'll get nonsensical result (the object itself).