[ Python ] shelve模块的使用

如果只需要存储一个简单的数据，shelve模块可以满足。
所要做的就是为它提供文件名。shelve最重要的函数是open，在调用它的时候（使用文件名作为参数），它会返回一个shelf对象，可以用它来存储内容。
只需要把它当成普通的字典就好。但是键一定要字符串，在完成工作之后，调用它的close方法。
1、潜在陷阱：
    意识到shelve.open函数返回的对象并不是普通的映射是很重要的，如下所示：
   

In [2]: s = shelve.open('test.db')

In [3]: s['x'] = ['a','b','c']

In [4]: s['x']
Out[4]: ['a', 'b', 'c']

In [5]: s['x'].append('d')

In [6]: s['x']
Out[6]: ['a', 'b', 'c']

（1）列表['a', 'b', 'c']存储在键x下
（2）获得存储的表示，并且根据它来创建新的列表，而'd'被添加到这个副本中，修改的版本还没有被保存。
（3）最终，再次获得原始版本——没有'd'

为了避免上面的问题，在使用shelve模块修改存储的对象，必须将临时变量绑定到获得的副本上，并且在它被修改后重新存储这个副本。

In [1]: import shelve

In [2]: s = shelve.open('test.db')

In [3]: s['x'] = ['a','b','c']

In [4]: s['x']
Out[4]: ['a', 'b', 'c']

In [5]: temp = s['x']        # 将副本绑定到临时变量上。

In [6]: temp.append('d')

In [7]: s['x'] = temp        # 再将临时变量写到副本上。

In [8]: s['x']
Out[8]: ['a', 'b', 'c', 'd']

2.简单的数据示例：

#!/usr/bin/python
import shelve
def store_person(db):    # 创建数据
    pid = raw_input('Enter unique ID number: ')     # 键值
    person = {}    # 定义person为字典
    person['name'] = raw_input('Enter name: ')    # 在person里定义子字典
    person['age'] = raw_input('Enter age: ')
    person['phone'] = raw_input('Enter phone: ')
    db[pid] = person    # 完成键值对应关系，保存到数据库
def lookup_person(db):
    pid = raw_input('Enter ID numbers: ')    # 获取键
    field = raw_input('What would you like to know ?(name,age,phone)')    # 获取子字典的键
    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('test.db')    # 创建shelve数据库
    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()

（1）以上代码将所有内容都放到函数中会让程序更加结构化。
（2）主程序放到main函数中，只有在if __name__ == '__main__'条件成立的时候才被调用。这意味着可以在其他程序中将这个程序作为模块导入，然后调用main()函数。
（3）在main函数中打开数据库，然后将其作为参数传给另外需要它的函数。在大多数情况下最好避免使用全局变量。
（4）对读取的内容调用strip和lower函数以生成了一个修改后的版本。如果总是对用户的输入使用strip和lower函数，那么就可以让用户随意输入大小写字母和添加空格
（5）使用try/finally确保数据库能正常关闭。我们永远不知道什么时候会出错，如果程序在没有正确关闭数据库的情况下终止，那么数据库文件就有可能被损坏。

运行结果如下：

[root@server1 python]# ./database1.py
Enter command(? for help): ?
The available commands are:
store : stores information about a person
lookup : looks up a person from ID number
quit : save changes and exit
? : prints this message
Enter command(? for help): store
Enter unique ID number: 002
Enter name: xixi
Enter age: 20
Enter phone: 133
Enter command(? for help): lookup
Enter ID numbers: 002
What would you like to know ?(name,age,phone)name
Name: xixi
Enter command(? for help): lookup
Enter ID numbers: 002
What would you like to know ?(name,age,phone)age
Age: 20
Enter command(? for help): quit

虽然普通字典也能达到这样的效果，但是shelve是将数据字典存储在数据库文件中，是存储在磁盘上，只要数据库文件不损坏，数据也不会丢失。