cwinfo/powerdns-admin
5
0
Fork 0
mirror of https://github.com/cwinfo/powerdns-admin.git synced 2024-11-08 14:40:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix 1329

Browse Source
This commit is contained in:
Aaron Carson 2022-12-14 00:34:12 +00:00
parent 9a38e1758f
commit ff671ebabe
1 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
powerdnsadmin/models/user.py
View File

@ -108,8 +108,9 @@ class User(db.Model):
def check_password(self, hashed_password): def check_password(self, hashed_password):
# Check hashed password. Using bcrypt, the salt is saved into the hash itself # Check hashed password. Using bcrypt, the salt is saved into the hash itself
if hasattr(self, "plain_text_password"): if hasattr(self, "plain_text_password"):
return bcrypt.checkpw(self.plain_text_password.encode('utf-8'), if self.plain_text_password != None:
hashed_password.encode('utf-8')) return bcrypt.checkpw(self.plain_text_password.encode('utf-8'),
hashed_password.encode('utf-8'))
return False return False
def get_user_info_by_id(self): def get_user_info_by_id(self):
@ -422,8 +423,12 @@ class User(db.Model):
self.role_id = Role.query.filter_by( self.role_id = Role.query.filter_by(
name='Administrator').first().id name='Administrator').first().id
self.password = self.get_hashed_password( if hasattr(self, "plain_text_password"):
self.plain_text_password) if hasattr(self, "plain_text_password") else '*' if self.plain_text_password != None:
self.password = self.get_hashed_password(
self.plain_text_password)
else:
self.password = '*'
if self.password and self.password != '*': if self.password and self.password != '*':
self.password = self.password.decode("utf-8") self.password = self.password.decode("utf-8")
@ -460,8 +465,9 @@ class User(db.Model):
# store new password hash (only if changed) # store new password hash (only if changed)
if hasattr(self, "plain_text_password"): if hasattr(self, "plain_text_password"):
user.password = self.get_hashed_password( if self.plain_text_password != None:
self.plain_text_password).decode("utf-8") user.password = self.get_hashed_password(
self.plain_text_password).decode("utf-8")
db.session.commit() db.session.commit()
return {'status': True, 'msg': 'User updated successfully'} return {'status': True, 'msg': 'User updated successfully'}
@ -476,9 +482,11 @@ class User(db.Model):
user.firstname = self.firstname if self.firstname else user.firstname user.firstname = self.firstname if self.firstname else user.firstname
user.lastname = self.lastname if self.lastname else user.lastname user.lastname = self.lastname if self.lastname else user.lastname
user.password = self.get_hashed_password(
self.plain_text_password).decode( if hasattr(self, "plain_text_password"):
"utf-8") if hasattr(self, "plain_text_password") else user.password if self.plain_text_password != None:
user.password = self.get_hashed_password(
self.plain_text_password).decode("utf-8")
if self.email: if self.email:
# Can not update to a new email that # Can not update to a new email that