def register():
if current_user.is_authenticated:
return redirect(url_for('main.home'))
form = RegistrationForm()
if form.validate_on_submit():
hashed_password = bcrypt.generate_password_hash(
form.password.data).decode('utf-8')
user = User(username=form.username.data,
email=form.email.data,
password=hashed_password)
db.session.add(user)
db.session.commit()
flash('Your account has been created! and you are now able to login',
'success')
return redirect(url_for('users.login'))
return render_template('register.html', title='Register', form=form)
def __init__(self, userObj):
"Initialize the user object"
self.user_name = userObj['user_name']
self.first_name = userObj['first_name']
self.last_name = userObj['last_name']
self.email = userObj['email']
self.password = bcrypt.generate_password_hash(
userObj['password']).decode('utf-8')
self.userObj = {
"user_name": self.user_name,
"first_name": self.first_name,
"last_name": self.last_name,
"email": self.email,
"password": self.password,
"date_created": datetime.datetime.utcnow()
}
def reset_token(token):
if current_user.is_authenticated:
return redirect(url_for('main.home'))
user = User.verify_reset_token(token)
if user is None:
flash('That is an invalid or expired token', 'warning')
return redirect(url_for('users.reset_request'))
form = ResetPasswordForm()
if form.validate_on_submit():
hashed_password = bcrypt.generate_password_hash(
form.password.data).decode('utf-8')
user.password = hashed_password
db.session.commit()
flash('Your password has been updated! and you are now able to login',
'success')
return redirect(url_for('users.login'))
return render_template('reset_token.html',
title='Reset Password',
form=form)