use config variable to identify admin account while it is created

This commit is contained in:
Joachim Lusiardi 2016-08-03 08:13:39 +02:00
parent 1e7db29de9
commit 44b92e58b5
1 changed files with 4 additions and 0 deletions

View File

@ -58,6 +58,10 @@ def user_registered_sighandler(app, user, confirm_token):
""" """
role = user_datastore.find_role('user') role = user_datastore.find_role('user')
user_datastore.add_role_to_user(user, role) user_datastore.add_role_to_user(user, role)
if user.email == app.config['ADMIN_MAIL']:
# if the user selected the preconfigured email for the admin account
role = user_datastore.find_role('admin')
user_datastore.add_role_to_user(user, role)
new_vehicle = Vehicle('default vehicle') new_vehicle = Vehicle('default vehicle')
db.session.add(new_vehicle) db.session.add(new_vehicle)
user.vehicles.append(new_vehicle) user.vehicles.append(new_vehicle)