shelve模块

shelve：

是一种简单的数据存储方案，他有一个有趣的函数就是open()，这个函数接收一个参数就是文件名，会返回一个shelf对象，你可以用他来存储内容，可以简单的把它当作一个字典，当你存储完毕的时候，调用它的close方法来关闭。

简单的使用shelve模块的数据库应用程序：

import sys, shelve

def store_person(db):
    """
    Query user for data and store it in the shelf object
    """

    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):
    """
    Query user for ID and desired filed.and fetch the corresponding data
    from the shelf object
    """

    pid = raw_input('Enter ID number: ')
    field = raw_input('What would you like to know?(name,age,phone)')
    field = field.strip().lower()
    print field.capitalize() + ':', db[pid][field]

def print_help():
    print 'The available commands are:'
    print 'store  : Stores information about a person'
    print 'lookup : Looks up a person from ID number'
    print 'quit   : Save changes and exit'
    print '?      : Prints this message'

def enter_command():
    cmd = raw_input('Enter command (? for help): ')
    cmd = cmd.strip().lower()
    return cmd

def main():
    database = shelve.open('F:\\database.dat')
    try:
        while(True):
            cmd = enter_command()
            if cmd == 'store':
                store_person(database)
            elif cmd == 'lookup':
                lookup_person(database)
            elif cmd == '?':
                print_help()
            elif cmd == 'quit':
                return
    finally:
        database.close()

if __name__ == '__main__': main()

代码中使用了try/finally来确保数据库能够正确关闭，在一些值进行读取之后，对读取的内容调用strip和lower函数，可以让用户随意输入大小写。

运行结果：

注意：

shelve是一个对象持久化保存方法，将对象保存到文件里面，一般来说对象的保存和恢复都是通过shelve来进行的。
千万不要自己创建.dat文件，如果自己创建的话，会提示 "db type could not be determined"
解决方法: 删除database.dat文件，首次运行后会自动生成该文件。