def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User()
user.username=form.username.data
user.email=form.email.data
user.firstName=form.firstName.data
user.lastName=form.lastName.data
user.ad_street=form.ad_street.data
user.ad_suburb=form.ad_suburb.data
user.ad_state=form.ad_state.data
user.ad_country=form.ad_country.data
if createUser(user,form.password.data)==True:
flash(Markup('<script>Notify("Congratulations, you are now a registered user!", null, null, "success")</script>'))
return redirect(url_for('auth.login'))
else:
flash(Markup('<script>Notify("Could not create account, please enter required fields.", null, null, "danger")</script>'))
return redirect(url_for('main.register'))
return render_template('register.html', title='Register', form=form)
def create_user():
form = RegistrationForm()
if current_user.is_authenticated:
if current_user.isAdmin:
if form.validate_on_submit():
user = User()
user.username=form.username.data
user.email=form.email.data
user.firstName=form.firstName.data
user.lastName=form.lastName.data
user.ad_street=form.ad_street.data
user.ad_suburb=form.ad_suburb.data
user.ad_state=form.ad_state.data
user.ad_country=form.ad_country.data
if createUser(user,form.password.data):
flash(Markup('<script>Notify("You have created a new user!", null, null, "success")</script>'))
return redirect(url_for('main.users'))
else:
flash(Markup('<script>Notify("Could not create account, please enter required fields.", null, null, "danger")</script>'))
else:
flash(Markup('<script>Notify("Only an admin user can view this page.", null, null, "danger")</script>'))
return redirect(url_for('main.index'))
return render_template('register.html', title='Register', form=form)
