def user(nickname):
user = User.filter_by(nickname = nickname).first()
if user == None:
flash('User' + nickname + ' not found.')
return redirect(url_for('index'))
posts = [
{'author': user, 'body': 'Test post #1'},
{'author': user, 'body': 'Test post #2'}
]
return render_template('user.html',
user = user,
posts = posts
)
def unfollow(nickname):
user = User.filter_by(nickname = nickname).first()
# check that user exists
if user == None:
flash('User '+ nickname +' not found.')
return redirect(url_for('index'))
# check that you aren't trying to un-follow yourself
if user == g.user:
flash('You can\'t unfollow yourself!')
return redirect(url_for('user', nickname = nickname))
# attempt to unfollow user
u = g.user.unfollow(user)
if u is None:
flash('Cannot unfollow '+ nickname +'.')
return redirect(url_for('user', nickname = nickname))
# write changes to db
db.session.add(u)
db.session.commit()
flash('You have stopped following '+ nickname +'.')
return redirect(url_for('user', nickname = nickname))
