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文件,首次运行后会自动生成该文件。