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)