def profile(monkey_id):
form = SignUpForm()
monkey = Monkey.query.get(monkey_id)
best_friend_name = 'Not yet'
bf_row = BestFriend.query.filter_by(monkey_id=monkey_id)
if bf_row.count():
best_friend_name = Monkey.query.get(bf_row.first().best_friend_id).name
friends_subquery = (Friendship.query
.filter_by(monkey_id=monkey_id)
.subquery())
friends = (Monkey.query.join(friends_subquery,
Monkey.monkey_id ==
friends_subquery.c.friend_id))
if form.validate_on_submit():
monkey.name = form.name.data
monkey.age = form.age.data
monkey.email = form.email.data
db.session.add(monkey)
db.session.commit()
flash('Monkey info edited')
return redirect(url_for('profile', monkey_id=monkey_id))
return render_template('profile.html',
title='Profile',
monkey=monkey,
monkey_id=monkey_id,
form=form,
best_friend_name=best_friend_name,
friends=friends)
def home():
form = SignUpForm()
if form.validate_on_submit():
monkey = Monkey(name=form.name.data,
age=form.age.data,
email=form.email.data)
db.session.add(monkey)
db.session.commit()
flash(monkey.name + ' came to life!')
return redirect(url_for('profile', monkey_id=monkey.monkey_id))
return render_template('home.html', title='Home', form=form)
