這是用戶登錄游戲的部分。該數(shù)據(jù)庫是在MLAB托管，有命名文檔的集合loginInfo定義mycol。我面臨的問題是，當(dāng)它返回 id 時，我無法檢查它是否有效，因為我無法在 Python 中檢查任何類型的本地列表。我如何解決它？import pymongodef login(): print('Login\nRegister\nPlay as Guest') login_input = input('What would you like to do?') if login_input == 'Login': username = input('Username:') username_fromDB = mycol.find({"username": username}) if username_fromDB == username: password = input('Password:') password_fromDB = mycol.find({"password": password}) if password == password_fromDB: print('Login success!\n') mainMenu() elif login_input == 'Register': new_username = input('Username:') new_password = input('Password:') tobeaddedtodb = {"username": new_username, "password": new_password} adding = mycol.insert_one(tobeaddedtodb) print("Registered!\n") elif login_input == 'Play as Guest': mainMenu()
1 回答
慕慕森
TA貢獻1856條經(jīng)驗 獲得超17個贊
username_fromDB = mycol.find({"username": username})
這是錯誤的，因為find()返回一個游標(biāo)對象，而不是您期望的用戶名。
假設(shè)該loginInfo集合包含唯一的用戶名和相應(yīng)的密碼，則無需find()對數(shù)據(jù)庫進行 2 次查詢。您可以只檢索用戶實體并檢查輸入提供的密碼是否與存儲在數(shù)據(jù)庫中的密碼匹配。
由于用戶名是唯一的，您應(yīng)該使用find_one()代替find()。如果用戶被成功找到它返回dict，否則你得到None。
嘗試這樣的事情：
import pymongo
def login():
print('Login\nRegister\nPlay as Guest')
login_input = input('What would you like to do?')
if login_input == 'Login':
username_inp = input('Username:')
pwd_inp = input('Password:')
user_found = mycol.find_one({"username": username_inp}) # query by specified username
if user_found: # user exists
if pwd_inp == user_found['password']:
print('Login success!\n')
mainMenu()
else:
print('Wrong password')
else:
print('User not found')
elif login_input == 'Register':
new_username = input('Username:')
new_password = input('Password:')
tobeaddedtodb = {"username": new_username, "password": new_password}
adding = mycol.insert_one(tobeaddedtodb)
print("Registered!\n")
elif login_input == 'Play as Guest':
mainMenu()
添加回答
舉報
0/150
提交
取消