def edit_profile():
user = User.query.get(session['user_id'])
form = EditProfileForm(obj=user)
form.populate_obj(user)
if form.validate_on_submit():
print user
db.session.add(user)
print "session add user"
db.session.commit()
print "session commit user"
return redirect(url_for('profile'))
return render_template('edit_profile.html', form=form)
def profile():
"""Update profile for current user."""
if not g.user:
flash("Access unauthorized.", "danger")
return redirect("/")
form = EditProfileForm(obj=g.user)
if form.validate_on_submit():
if User.authenticate(g.user.username, form.password.data):
flash("Attaboy", "success")
form.populate_obj(g.user)
db.session.commit()
else:
flash("Invalid password dipstick.", "danger")
return redirect('/users/profile')
return redirect("/")
# IMPLEMENT THIS
return render_template("users/edit.html", form=form, user=g.user)
