def update_user(self):
user = User.find_by_email(current_user.email)
user.username = self.username.data
user.email = self.email.data
if self.picture.data:
picture_file = save_picture(self.picture.data)
current_user.image_file = picture_file
user.save_to_db()
flash('Your account has been updated!')
def login():
form = LoginForm()
if form.validate_on_submit():
user = User.find_by_email(form.email.data)
if user and bcrypt.check_password_hash(user.password,
form.password.data):
login_user(user, remember=form.remember.data)
next_page = request.args.get('next')
return redirect(next_page) if next_page else redirect(
url_for('main.home'))
else:
flash('Login unsuccessful. Please check username & password',
'danger')
return render_template('login.html', title='Login', form=form)
def validate_email(self, email):
if email.data != current_user.email:
user = User.find_by_email(email.data)
if user:
raise ValidationError('That email is already taken.')
