def add_restaurant():
"""Allows user to add restaurant to a list"""
item_id = request.form.get('id')
restaurant_name = request.form.get('restaurant_name')
yelp_rating = request.form.get('yelp_rating')
latitude = request.form.get('latitude')
longitude = request.form.get('longitude')
list_id = request.form.get('list_id')
address = request.form.get('address')
categories = request.form.get('categories')
neighborhoods = request.form.get('neighborhoods')
link = request.form.get('url')
# check if restaurant already in db
get_restaurant = Restaurant.query.filter_by(restaurant_name=restaurant_name, latitude=latitude, longitude=longitude).first()
if get_restaurant:
# check if already part of list then add if no
get_restaurant_list = RestaurantList.query.filter_by(restaurant_id=get_restaurant.restaurant_id, list_id=list_id).first()
if get_restaurant_list:
flash("Restaurant is already part of list")
else:
new_restaurant_list = RestaurantList(restaurant_id=get_restaurant.restaurant_id, list_id=list_id)
db.session.add(new_restaurant_list)
db.session.commit()
# if restaurant is not already in db, add it and add to RestaurantList
else:
new_restaurant = Restaurant(restaurant_name=restaurant_name, yelp_rating=yelp_rating, latitude=latitude, longitude=longitude, address=address, categories=categories, neighborhoods=neighborhoods, link=link)
db.session.add(new_restaurant)
db.session.commit()
# need this line because we just added the restaurant to db and need to get the id to add to RestaurantList
restaurant_info = Restaurant.query.filter_by(restaurant_name=restaurant_name, latitude=latitude, longitude=longitude).first()
new_restaurant_list = RestaurantList(restaurant_id=restaurant_info.restaurant_id, list_id=list_id)
db.session.add(new_restaurant_list)
db.session.commit()
return jsonify(status='success', id=item_id, restaurant_name=restaurant_name, yelp_rating=yelp_rating, latitude=latitude, longitude=longitude)
评论列表
文章目录