Python从pickle保存的d中删除条目

2024-06-08 22:42:54 发布

您现在位置:Python中文网/ 问答频道 /正文

首先,我知道this post哪个状态是我必须重写整个文件才能从pickle保存的数据中删除1个项。

我有一个以二进制形式保存用户名和密码(密码的哈希值)的文件。它由以下代码创建:

import pickle
import hashlib

def Encryption(data):
    return hashlib.sha224(data).hexdigest()

db = {'user1' : Encryption('password1'), 'user2' : Encryption('password2'), 'user3' : Encryption('password3')}
fh = open('database.db', 'wb')
pickle.dump(db, fh)
fh.close()

我想从文件中删除user2,password2(第二个条目)。所以这就是我所做的

import pickle
import hashlib
from os import path

n='user2'

def Encryption(data):
    return hashlib.sha224(data).hexdigest()

if path.isfile('database.db'):
    fh=open('database.db','rb')
    db=pickle.load(fh)
    fh.close()

_newlist,_newlist2=([] for i in range (2))    
_username=[]
_password=[]

#Get the user names and passwords hash values into two different list
for user in db:
    _username.append(user)
    _password.append(db[user])    

#If user name is equal to the user name i want to delete, skip . Else append it to new list
for i in range(len(_username)):
    if n==_username[i]:
        pass
    else:
        _newlist.append(_username[i])
        _newlist2.append(_password[i])

#Clear the file
fh=open('database.db','wb')
fh.close()

#Re-write the new lists to the file
for i in range(len(_newlist)):
    db={_newlist[i]:_newlist2[i]}
    fh = open('database.db', 'ab')
    pickle.dump(db,fh)

它不删除第二个条目(user2,password2),而是删除除最后一个条目以外的所有条目。有人帮我指出我的代码有什么问题吗?


Tags: theimportdbdatausername条目opendatabase