def createUser(flask_session):
newUser = User(
name=flask_session["username"],
picture=flask_session["picture"],
link=flask_session["link"],
email=flask_session["email"],
google_plus_id=flask_session["google_plus_id"],
)
session.add(newUser)
session.commit()
user = session.query(User).filter_by(google_plus_id=flask_session["google_plus_id"]).one()
return user.id
def newRestaurant():
"""page to create a new menu item."""
if "access_token" not in flask_session:
return logInRedirect()
user_id = getUserId(flask_session["email"], flask_session["google_plus_id"])
if request.method == "POST":
new_name = request.form["new_name"]
print "\nnewRestaurant POST triggered, name is: ", new_name
newRestaurant = Restaurant(name=new_name, user_id=user_id)
session.add(newRestaurant)
session.commit()
flash("new restaurant '" + new_name + "' created!")
print "POST worked!"
return redirect(url_for("showRestaurants"))
else:
return render_template("newRestaurant.html")
def newMenuItem(restaurant_id):
"""page to create a new menu item."""
if 'access_token' not in flask_session:
return logInRedirect()
restaurant = session.query(Restaurant).filter_by(id = restaurant_id).first()
user_id = getUserId(flask_session['email'],flask_session['google_plus_id'])
if not restaurant.user_id == user_id:
flash("Only restaurant owners can add new items.")
return redirect(url_for("publicMenu",restaurant_id = restaurant_id))
if request.method == "POST":
new_name = request.form['new_name']
print "\nnewMenuItem POST triggered, name is: ", new_name
newMenuItem = MenuItem( name=new_name,
restaurant_id=restaurant.id )
session.add(newMenuItem)
session.commit()
flash( "new item '" + new_name + "' created!")
print "POST worked!"
return redirect(url_for("showMenu", restaurant_id=restaurant.id))
else:
return render_template('newMenuItem.html', restaurant = restaurant)
