def validate_form(self, form):
"""
Validate the form on submit.
:param form:
Form to validate
"""
return helpers.validate_form_on_submit(form)
def login_view(self):
# handle user login
form = LoginForm(request.POST)
if helpers.validate_form_on_submit(form):
user = form.get_user()
login.login_user(user)
if login.current_user.is_authenticated():
return redirect(url_for(".index"))
link = "<p>Don't have an account? <a href=\"" + url_for(".register_view") + '">Click here to register.</a></p>'
self._template_args["form"] = form
self._template_args["link"] = link
return super(MyAdminIndexView, self).index()
def register_view(self):
form = RegistrationForm(request.POST)
if helpers.validate_form_on_submit(form):
user = User()
form.populate_obj(user)
# we hash the users password to avoid saving it as plaintext in the db,
# remove to use plain text:
user.password = generate_password_hash(form.password.data)
db.session.add(user)
db.session.commit()
login.login_user(user)
return redirect(url_for(".index"))
link = '<p>Already have an account? <a href="' + url_for(".login_view") + '">Click here to log in.</a></p>'
self._template_args["form"] = form
self._template_args["link"] = link
return super(MyAdminIndexView, self).index()
