def add_to_cart():
if request.form:
try:
product = get_or_404(Product, request.form['product_id'])
except KeyError:
return redirect(request.referrer)
shoppingcart = ShoppingCart.for_session_cart(request.cart)
form = AddToCartForm(request.form, product=product, cart=shoppingcart)
if form.validate():
flash('Added {0} {1} to the cart'.format(
form.quantity.data, product.name))
form.save()
return redirect(url_for('.index'))
return redirect(request.referrer)
def index():
shoppingcart = ShoppingCart.for_session_cart(request.cart)
forms = [ReplaceCartLineForm(request.form, product=product,
cart=shoppingcart, product_id=product.id_) for product in shoppingcart.items]
return render_template('cart.html', cart=shoppingcart, forms=forms)
