您的位置:首页 > 编程语言 > Python开发

Python使用shelve模块实现简单数据存储的方法

2015-05-20 15:46 1151 查看

本文实例讲述了Python使用shelve模块实现简单数据存储的方法。分享给大家供大家参考。具体分析如下:

Python的shelve模块提供了一种简单的数据存储方案,以dict(字典)的形式来操作数据。

#!/usr/bin/python
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 field,
and fetch the corresponding data
from the shelf object
"""
pid = raw_input('Enter unique ID number:')
temp = db[pid]
field = raw_input('Please enter name, age or phone:')
field.strip().lower()
print field.capitalize() + ': ', temp[field]
def print_help():
print 'The avaliable commands are:'
print 'store  :Stores infomation about a person'
print 'lookup  :Looks up a person form 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('database')
# database stores in current directory
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()
# Close database in any condition
if __name__ == '__main__':
main()

希望本文所述对大家的Python程序设计有所帮助。

您可能感兴趣的文章:

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  Python shelve 数据存储