Python简要标准库（3）

shelve

---

若只需要一个简单的存储方案，那么shelve模块可以满足你大部分的需要，你所需要的只是为它提供文件名。shelve中唯一有趣的函数是open，在调用的时候他会返回一个Shelf对象

注意：

　　只需要把它当作普通字典（但是键一定要是字符串）来操作

　　在操作完之后，调用它的close方法

在python2.4之后的版本还有一个解决方案，将open函数的writeback参数设为True。这样所有从shelf读取或者赋值到shelf的数据结构都会保存在内存(缓存)中，而且只有在关闭shelf的时候才会回写到硬盘。

open函数参数说明

这个主要是解决这个问题：

 import shelve
 s = shelve.open('data.dat','c',None,False)
 s['x'] = ['a','b','c']
 s['x'].append('d')
 print (s['x'])

会发现 s['x']还是'a','b','c' 这是因为在s['x'].append('d') 有一个创建副本的过程，即先会提取s['x'] 的内容建立一个副本，然后会在这个副本上追加'd',但此时的变更不会同步在s['x']中。

所以解决办法阔以这样

 import shelve
 s = shelve.open('data.dat','c',None,False)
 s['x'] = ['a','b','c']
 temp = s['x']              ！
 temp.append('d')           ！
 s['x'] = temp              ！
 print (s['x'])

这段的亮点主要在s['x'] = temp  等同于一个回写的操作，这样就把变更后的数据写入了s['x']

示例

最后有一段个人信息录入/查看程序

　　*增加了在添加信息时测试是否重复的过程

 import sys,shelve
 2 #储存信息
 def store_person(db):
         while (True):
                 pid = input('Enter unique ID number : ')
                 if pid in db.keys():
                         print ('find the same key : ')
                 else:
                         break

         person = {}
         person ['name'] = input('Enter name : ')
         person ['age']  = input('Enter age : ')
         person ['phone'] = input('Enter phone number : ')

         db[pid] = person
         print ('add to db done!')
 #查找信息
 def lookup_person(db):
     pid = input('Enter ID number : ')
     field = input('What would you like to know?(name,age,phone,all) : ')
     field = field.strip().lower()
     if field == 'all':
         d = db[pid]
         for key,value in d.items():
                 print (key,':',value)
     else:
         print (field.capitalize() + ':',db[pid][field])
 #。。。
 def print_help():
     print ('help something')

 #获取用户输入
 def enter_command():
     cmd = input('Enter command(? for help) : ')
     cmd = cmd.strip().lower()
     return cmd

 def main():
     database = shelve.open('date.dat')
     try:
         while True:
             cmd = enter_command()
             if cmd == 's':
                 store_person(database)
             elif cmd == 'l':
                 lookup_person(database)
             elif cmd == '?':
                 print_help()
             elif cmd == 'q':
                 break;
     finally:
             database.close()
             print ('stop by user')

 if  __name__ == '__main__' : main()