Python醃密碼

Question

這是我使用pickle的python密碼系統。我知道這很糟糕，但這是我第一次用泡菜。

import pickle 
import os 

userlist = {'user1':'userpass1', 'user2':'userpass2'} 

users = open ("users.pkl", 'wb') 

pickle.dump (userlist, users) 

username = input ("Enter your username: ") 
password = input ("Enter your password: ") 

if (username in userlist) and (password == userlist[username]): 
    print ("Access Granted") 
else: 
    newaccount = input ("User not found. Shall I create a new account? ") 
    if newaccount == "yes": 
     username = input ("Please enter your username: ") 
     password = input ("Please enter yout password: ") 
     userlist.update({username:password}) 
     pickle.dump (userlist, users) 
     users.close() 

我的問題是，每當我去添加一個新的賬戶，使用該部分：

newaccount = input ("User not found. Shall I create a new account? ") 
    if newaccount == "yes": 
     username = input ("Please enter your username: ") 
     password = input ("Please enter yout password: ") 
     userlist.update({username:password}) 
     pickle.dump (userlist, users) 
     users.close() 

它似乎增加它（和它看起來像它的存在在使用泡菜文件記事本）但是，我重新啓動python文件，並沒有看到它。

我相信這是值得做這部分：

userlist = {'user1':'userpass1', 'user2':'userpass2'} 

users = open ("users.pkl", 'wb') 

pickle.dump (userlist, users) 

任何幫助表示讚賞！ ：d

Answer 1

你每次與w運行程序時覆蓋：

users = open ("users.pkl", 'wb') 

如果你想獲得你需要看到，如果該文件已經存在，先前醃項目和pickle.load獲得先前酸菜項目，然後在您的代碼末尾輸入dump。

類似以下內容：

from tempfile import NamedTemporaryFile 

try: 
    # see if we have run this before 
    with open ("users.pkl", 'rb') as users: 
     users_dict = pickle.load(users) 
except IOError: 
    # if not set to defualt 
    users_dict = {'user1':'userpass1', 'user2':'userpass2'} 


username = input ("Enter your username: ") 
password = input ("Enter your password: ") 

if users_dict.get(username) == password: # unless a password can be None we can use get 
    print ("Access Granted") 
else: 
    newaccount = input("User not found. Shall I create a new account? ") 
    if newaccount == "yes": 
     username = input("Please enter your username: ") 
     password = input ("Please enter yout password: ") 
     users_dict[username] = password # just use key = value 

with NamedTemporaryFile("wb",dir=os.path.dirname("users.pkl"),delete=False) as f: # in case we get exception use temp file 
    pickle.dump (users_dict, f) 
os.replace(f.name,"users.pkl") # update original