#!/usr/bin/python

###############################
#
# This program will track things that you have loaned to your friends
#
#

import string

# Function definitions

# Draw menu

def menu():
    print
    print "1. Add person"
    print "2. Add item to person"
    print "3. Remove item from person"
    print "4. Remove person"
    print "5. View all loaned items"
    print "9. Save data and exit"
    print

# End function definitions

dbase = {}

try:                        # Open the data file and import it
    inFile = open("data.txt", "r")
    while 1:
        inLine = inFile.readline()
        if len(inLine) == 0:
            break
        temp = inLine.split(',')
        dbase[temp[-2]] = temp[:-2]
        
    inFile.close()
except IOError:             # If no file found, we'll make a new one
        print "No saved data found.  Creating new file..."
    
choice = "0"
while choice != "9":        # main menu loop
    menu()
    choice = raw_input("Selection:")
    if choice == "1":       # add person
        name = raw_input("Enter name:")
        dbase[name] = []
        

    elif choice == "2":     # add item
        name = raw_input("Person's name:")
        if dbase.has_key(name):
            item = raw_input("Item:")
            dbase[name].append(item)
        else:
             print name, "was not found."


    elif choice == "3":     # remove item
        name = raw_input("Enter name from which to remove item:")
        if dbase.has_key(name):
            item = raw_input("Enter item to remove:")
            if item in dbase[name]:
                number = dbase[name].index(item)
                del dbase[name][number]
            else:
                print "Item not found."
        else:
            print "Name not found."


    elif choice == "4":     # remove person
        name = raw_input("Enter name to delete:")
        if dbase.has_key(name):
            if not len(dbase[name]) == 0:
                print name, "still has items on loan."
                certain = raw_input("Delete anyway? (y/N):")
                if certain == "y" or certain == "Y" or certain == "yes":
                    del dbase[name]
                else:
                    print ""
            else:
                del dbase[name]
        else:
             print name, "was not found."


    elif choice == "5":     # view all items
         for x in dbase.keys():
            print "%s:" % (x),
            for y in dbase[x]:
                if dbase[x].index(y) != len(dbase[x]) -1:
                    print "%s," % (y),
                    
                else:
                    print y
            print""

print ""

# Now we save the file
try:
    outFile = open("data.txt", "w")
    for x in dbase.keys():
        dbase[x].append(x)
        for y in dbase[x]:
            output = y
            output += ","
            outFile.write(output)
        outFile.write("\n")
    outFile.close()
    print "Data saved..." 

except NameError:
    print "No data saved..."

try:
    file = open("data.txt")
    for line in file:
        if line == "": break
        values = line.split(',')
        dbase[values[-2]] = values[:-2]
finally:
    file.close()

import csv

try:
    file = open("data.txt", "rb")
    for row in csv.reader(file):
        if not row: break
        dbase[rows[-2]] = rows[:-2]
finally:
    file.close()

def add_person():
    name = raw_input("Enter name:")
    dbase[name] = []

...

choices = {
    0 : add_person,
    1 : add_item,
    ...
}

while choice != 9:
    print menu
    try:
       choice = int(raw_input("Selection:"))
       choices[choice]()
    except ValueError:
       print "Invalid choice: must be a number"
    except KeyError:
       print "Invalid choice: number must be from 0 to 9"

import shelve

dbase = shelve.open("data.shf")

         for x in dbase.keys():
            print "%s:" % (x),
            for y in dbase[x]:
                if dbase[x].index(y) != len(dbase[x]) -1:
                    print "%s," % (y),
                    
                else:
                    print y
            print""

for key, list in dbase.items():
    print key + ":"
    print ",".join(list)
print