[Python]shelve模块实例一则

最近在看到python的shelve模块，借鉴网上的脚本自己修改写了一个实例。 实例说明 实例中，数据写入到和代码同级的database文件之中。 文件数据不能累加写入，每次写入前都会覆盖前面的数据，也就是说只能保存上一次的数据。

import sys, shelve

def store_person(db):
     pid = raw_input('Enter unique ID number: ')
     person = {}
     person['name'] = raw_input('Enter name: ')
     person['age'] = raw_input('Enter age: ')
     person['phone'] = raw_input('Enter phone number: ')
     db[pid] = person

def lookup_person(db):
     pid = raw_input('Enter unique ID number: ')
     temp = db[pid]
     print """ID '%s' info:
        Name: '%s'
        age: '%s'
        phone: '%s'
        """ % (pid,temp['name'],temp['age'],temp['phone'])

Menu = {'r':store_person, 'c':lookup_person}

def main():
    ShowMenu = """Enter your choice:
    (R)egist user info
    (C)heck user info
    (Q)uit
    """
    while True:
        while True:
            try:
                choice = raw_input(ShowMenu).strip()[0].lower()
            except(EOFError,KeyboardInterrupt,IndexError):
                choice = 'q'
            print "\nYou picked:['%s']" % choice

            if choice not in 'rcq':
                print 'invalid option, please try again'
            else:
                break
        if choice == 'q':
            break
        PersonInfo = shelve.open('database')
        Menu[choice](PersonInfo)
        PersonInfo.close()

if __name__ == '__main__':
    main()