Dictionary with classes?

Question

In Python is it possible to instantiate a class through a dictionary?

shapes = {\'1\':Square(), \'2\':Circle(), \'3\':Triangle()}

x = shapes[raw_input()]


        

Answer 1

Almost. What you want is

shapes = {'1':Square, '2':Circle, '3':Triangle} # just the class names in the dict

x = shapes[raw_input()]() # get class from dict, then call it to create a shape instance.

Answer 2

I'd recommend a chooser function:

def choose(optiondict, prompt='Choose one:'):
    print prompt
    while 1:
        for key, value in sorted(optiondict.items()):
            print '%s) %s' % (key, value)
        result = raw_input() # maybe with .lower()
        if result in optiondict:
            return optiondict[result]
        print 'Not an option'

result = choose({'1': Square, '2': Circle, '3': Triangle})()