好的,我们可以这样完善你的登录代码来处理用户不存在的情况。
username =input("请输入用户名: ")
password = input("请输入密码: ")
# 从数据库中查询用户信息
user = db.query_user(username)
if user is None:
# 用户不存在
print("用户不存在")
else:
# 用户存在,检查密码是否正确
if user.password == password:
# 登录成功
print("登录成功")
else:
# 登录失败,密码错误
print("密码错误")
在上面的代码中,我们首先使用 db.query_user(username)
从数据库中查询用户信息,如果返回值为 None
,则说明用户不存在,否则说明用户存在,我们就可以检查密码是否正确了。
希望这对你有帮助。如果你还有任何疑问,请随时联系我。