EXIT = 9 while selection != EXIT: print "Enter your selection:" print "1. Squirrel login" print "2. Hide Acorn" print "3. Retrieve Acorn" print "4. Write squirrels to file" print "5. Write acorns to file" print "9. Exit" while selection < 1 or (selection > 5 and selection != EXIT): selection = input (name + "> ") if selection == 1: print "Squirrel login..." squirrel = Squirrel() name = squirrel.make_name() if name in squirrels: squirrel = squirrels[name] print "Returning squirrel has acorns at:" for acorn in squirrel.acorns: acorn.show() print "Ready to retrieve or add new acorns..." else: squirrels[name] = squirrel print "New squirrel. Ready to add new acorns..." elif selection == 2: print "Hide acorn..." new_acorn = Acorn() print name new_acorn.set_owner(name) squirrel.hide_acorn(new_acorn)