server.py 文件源码

def add_restaurant_to_faves():
    """Adds a restaurant to a user's favorites list"""

    restaurant_id = request.form.get('rest_id')
    user_id = session.get('user')

    get_fave_restaurant = Fave.query.filter_by(restaurant_id=restaurant_id, user_id=user_id).first()

    if get_fave_restaurant:
        Fave.query.filter_by(fave_id=get_fave_restaurant.fave_id).delete()
        db.session.commit()
    else:
        fave_restaurant = Fave(restaurant_id=restaurant_id, user_id=user_id)
        db.session.add(fave_restaurant)
        db.session.commit()

    return jsonify(result="success", id=restaurant_id)