文章的关键点:将函数open的参数writeback设置为True就不需要引用一个临时变量。
开始文章的叙述:
如何将数据存储到文件中,如果需要的是简单的存储方案,模块shelve可替你完成大部分工作——你只需提供一个文件名即可。对于模块shelve,你唯一感兴趣的是函数open。这个函数将一个文件名作为参数,并返回一个Shelf对象,供你用来存储数据。你可像操作普通字典那样操作它(只是键必须为字符串),操作完毕(并将所做的修改存盘)时,可调用其方法close。该模块作用用于创建永久性映射,其内容存储在使用给定文件名的数据库中。
1.一个潜在的陷阱
至关重要的一点是认识到shelve.open返回的对象并非普通映射,如下例所示:
i
import shelve
s = shelve.open('test.dat')
s['x'] = ['a', 'b', 'c']
s['x'].append('d')
s['x']
结果:['a', 'b', 'c']
'd’到哪里去了呢?
这很容易解释:当你查看shelf对象中的元素时,将使用存储版重建该对象,而当你将一个元素赋给键时,该元素将被存储。在上述示例中,发生的事情如下。
列表[‘a’, ‘b’, ‘c’]被存储到s的’x’键下。
获取存储的表示,并使用它创建一个新列表,再将’d’附加到这个新列表末尾,但这个修改后的版本未被存储!
最后,再次获取原来的版本——其中没有’d’。
要正确地修改使用模块shelve存储的对象,必须将获取的副本赋给一个临时变量,并在修改这个副本后再次存储:
temp = s['x']
temp.append('d')
s['x'] = temp
s['x']
结果:['a', 'b', 'c', 'd']
还有另一种避免这个问题的办法:将函数open的参数writeback设置为True。这样,从shelf对象读取或赋给它的所有数据结构都将保存到内存(缓存)中,并等到你关闭shelf对象时才将它们写入磁盘中。如果你处理的数据不多,且不想操心这些问题,将参数writeback设置为True可能是个不错的主意。在这种情况下,你必须确保在处理完毕后将shelf对象关闭。为此,一种办法是像处理打开的文件那样,将shelf对象用作上下文管理器。