wsy182
/
SmartRollCall
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更新 database_manager.py

master
wangsiyuan 2023-12-25 21:43:36 +08:00
parent 6026eac50e
commit 08790e3e9e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
db/database_manager.py
View File

@ -46,17 +46,18 @@ class DatabaseManager:
def valid_login(self, phone_number, password_attempt): def valid_login(self, phone_number, password_attempt):
# SQL查询获取用户的哈希密码，身份和状态 # SQL查询获取用户的哈希密码，身份和状态
sql = "SELECT password, identity, status FROM user WHERE phone_number=%s LIMIT 1" sql = "SELECT password, identity, status,nickname FROM user WHERE phone_number=%s LIMIT 1"
result = self.fetch(sql, (phone_number,)) result = self.fetch(sql, (phone_number,))
if result: if result:
stored_hash = result[0]['password'] # 假设结果是密码字段 stored_hash = result[0]['password'] # 假设结果是密码字段
identity = result[0]['identity'] # 用户身份 identity = result[0]['identity'] # 用户身份
status = result[0]['status'] # 用户状态 status = result[0]['status'] # 用户状态
nickname = result[0]['nickname']
# 使用bcrypt进行密码验证 # 使用bcrypt进行密码验证
if bcrypt.checkpw(password_attempt.encode('utf-8'), stored_hash.encode('utf-8')): if bcrypt.checkpw(password_attempt.encode('utf-8'), stored_hash.encode('utf-8')):
# 密码匹配，返回登录成功，身份和状态 # 密码匹配，返回登录成功，身份和状态
return {'valid': True, 'identity': identity, 'status': status} return {'valid': True, 'identity': identity, 'status': status,'nickname': nickname}
# 密码不匹配或用户不存在，返回登录失败 # 密码不匹配或用户不存在，返回登录失败
return {'valid': False} return {'valid': False}