版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/yysyangyangyangshan/article/details/85084999
如果我们只需要存少量的数据,一个简单的存储方案是使用shelve模版。我们只需要给他提供一个指定路径的文件名。shelve调用过程是,
先调用open函数,参数为文件名。返回值是一个shell对象。这个对象可以用来存储数据,可以作为一个字典来操作,但是键一定是字符串类型的。操作完成后调用close函数。看下shelve的简单使用,
import shelve
#参数为文件名,会在当前运行目录下创建三个文件
#test.dat.bak
#test.dat.dat
#test.dat.dir
s=shelve.open('test.dat')
s['x'] = ['a','b','c']
s['x'].append('d')
print(s['x'])
输出
['a', 'b', 'c']
temp=s['x']
temp.append('d')
s['x']=temp
print(s['x'])
输出
['a', 'b', 'c', 'd']
第一段代码中,d不见了,因为当用shell对象查找元素时,这个对象根据已经存储的版本进行构建,当给某个键时,被存储了。
上述过程如下,
['a','b','c']赋值给x键;
d添加时,当前版本并未保存;
获取值时,获取的是原始版本,因此d不见了。
而第二段,temp是副本,副本修改后,重新保存了,所以能获取到d。我们再通过这个例子,看看shelve实现简单的数据存储。
如下,使用shelve模版模拟简单数据库的使用。
import sys,shelve
def store_person(db):
"""
Query user for data store it in the shelf object
"""
pid=input("输入唯一编码:")
person={}
person['name'] =input('请输入姓名:')
person['age']=input('请输入年龄:')
person['phone']=input('请输入电话号码:')
db[pid]= person
def lookup_person(db):
"""
Query user for id and desired field,and fetch the corresponding data form the shelf object
"""
pid = input('输入唯一编码:')
field=input('以下信息请输入:(name,age,phone)')
field=field.strip().lower()
print(field.capitalize()+':',\
db[pid][field])
def print_help():
print('有效的命令有:')
print('store: 保存一个person对象')
print('lookup: 通过id查询一个person对象')
print('quit: 保存退出')
print('?:显示帮助信息')
def endter_command():
cmd=input('输入命令:(按?请求帮助)')
cmd=cmd.strip().lower()
return cmd
def main():
#此处路径根据实际的文件路径更改,不用提前创建该文件,会自动创建
#D:\\work\\Python\\是我本地电脑存在的目录,大家测试可以自行修改目录
database= shelve.open('D:\\work\\Python\\database.dat')
try:
while True:
cmd=endter_command()
if cmd=='store':
store_person(database)
elif cmd=='lookup':
lookup_person(database)
elif cmd=='?':
print_help()
elif cmd =='quit':
return
finally:
database.close()
#调用
main()
操作输出如下
输入命令:(按?请求帮助)?
有效的命令有:
store: 保存一个person对象
lookup: 通过id查询一个person对象
quit: 保存退出
?:显示帮助信息
输入命令:(按?请求帮助)store
输入唯一编码:1
请输入姓名:yangyoushan
请输入年龄:31
请输入电话号码:13011111
输入命令:(按?请求帮助)lookup
输入唯一编码:1
以下信息请输入:(name,age,phone)name
Name: yangyoushan
输入命令:(按?请求帮助)quit
功能文件下载:https://download.csdn.net/download/yysyangyangyangshan/10860069