使用pickle在python中保存字典

Question

我有一个文件hash_db.pickle，当我创建它时，我在上面保存了一本字典：

v = {hash_value:{"file name":file_name,"file size":file_size,"last scanned time":scanned_time}}

{123dfre345:{"file name":calc.pdf,"file size":234,"last scanned time":12:23 24/12/2013}}
{3gcdshj754:{"file name":star.pdf,"file size":10,"last scanned time":10:30 10/10/2013}}

所以如果我只想从文件中更改last scanned time 为3gcdshj754

我该怎么做？

Answer 1

使用pickle很简单，写的时候用

dct = {'3gcdshj754': {'file name': 'star.pdf', 'last scanned time': '10:30 10/10/2014', 'file size': '10'}}

import pickle
pickle.dump(dct, open("save.p", "wb"))

然后，在阅读时，使用

import pickle
dct_read = pickle.load(open("save.p", "rb"))

请注意，无论何时，您都必须以二进制模式 (b flag) 打开文件。

现在编辑内容很简单：

dct_read.values()[0]["last scanned time"] = '10:10 10/10/2010'

---

或者，正如@mhawke 在his answer 中建议的那样，您可以使用shelve。

Answer 2

您可以使用pickle。

import pickle
d = pickle.load(open('hash_db.pickle', 'rb'))
d['3gcdshj754']['last scanned time'] = '11:30 11/10/2015'
pickle.dump(d, open('hash_db.pickle', 'wb'))

但您可能会发现shelve 模块比直接使用pickle 更方便一些。它提供了一个持久字典，这似乎正是您想要的。示例用法：

import shelve
from datetime import datetime, timedelta

# create a "shelf"
shelf = shelve.open('hash_db.shelve')
shelf['123dfre345'] = {"file name": 'calc.pdf', "file size": 234, "last scanned time": datetime(2013, 12, 24, 12, 23)}
shelf['3gcdshj754'] = {"file name": 'star.pdf', "file size": 10, "last scanned time": datetime(2013, 10, 10, 10, 30)}
shelf.close()

# open, update and close
shelf = shelve.open('hash_db.shelve')
file_info = shelf['3gcdshj754']
file_info['last scanned time'] += timedelta(hours=+1, minutes=12)
shelf['3gcdshj754'] = file_info
shelf.close()

就是这样。