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()