def add_favorite():
form = FavoriteForm()
form['csrf_token'].data = request.cookies['csrf_token']
if form.validate_on_submit():
fav = Favorite(user_id=form.user_id.data,
listing_id=form.listing_id.data)
db.session.add(fav)
db.session.commit()
return fav.to_dict()
else:
return {"errors": "invalid favorite submission"}
def fav_drink(id, drink_id):
if request.method == 'POST':
# Create, add, and return new favorite
new_favorite = Favorite(user_id=id, drink_id=drink_id)
db.session.add(new_favorite)
db.session.commit()
return {'new_favorite_id': new_favorite.to_dict()['drink_id']}
if request.method == 'DELETE':
# Find and remove targeted favorite
favorite_to_delete = Favorite.query.filter(
and_(Favorite.user_id == id, Favorite.drink_id == drink_id)).one()
db.session.delete(favorite_to_delete)
db.session.commit()
return {}
def setFavorite(flipId):
favorite = Favorite(userId=current_user.id, flipId=flipId)
db.session.add(favorite)
db.session.commit()
return favorite.to_dict()
