少说话多写代码之Python学习063——标准模块(shelve模块)

版权声明:本文为博主原创文章,未经博主允许不得转载。 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

猜你喜欢

转载自blog.csdn.net/yysyangyangyangshan/article/details/85084999