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)
