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

Change to promote first register user to Administrator

Browse Source
This commit is contained in:
Khanh Ngo 2015-12-17 22:35:04 +07:00
parent 1e8bc55f36
commit 9d65e2ad17
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
app/models.py
View File

@ -180,7 +180,12 @@ class User(db.Model):
self.firstname = self.username
self.lastname = ''
self.role_id = 2
# first register user will be in Administrator role
if User.query.count() == 0:
self.role_id = 1
else:
self.role_id = 2
self.create_user()
logging.info('Created user "%s" in the DB' % self.username)
return True
@ -218,7 +223,12 @@ class User(db.Model):
return 'Email already existed'
try:
self.role_id = 2
# first register user will be in Administrator role
if User.query.count() == 0:
self.role_id = 1
else:
self.role_id = 2
user = User(username=self.username, firstname=self.firstname, lastname=self.lastname, role_id=self.role_id, email=self.email, password=self.get_hashed_password(self.plain_text_password))
db.session.add(user)
db.session.commit()