def fav(bid):
if current_user.is_anonymous():
flash(u'请先登录帐号，再收藏小说', 'error')
return (
redirect(url_for('login', target=url_for('book', bid=bid)))
)
else:
action = request.form.get('action', 'fav')
refer = request.form.get('refer', '')
if action == 'fav':
Favourite.add(current_user.id, bid)
else:
Favourite.remove(current_user.id, bid)
if refer == 'user_page':
return redirect(url_for('user', uid=current_user.id))
return redirect(url_for('book', bid=bid))
