def add_to_cart(request):
try:
sel_prod_id = request.POST['prod_id']
except:
return render(request, 'index.html', {
'err_msg': 'Product selecting error.'
})
product = get_object_or_404(Product, pk=sel_prod_id)
cart = Cart(request)
cur_q = cart.get_current_quantity(product)
if cur_q == 0:
cart.add(product, product.price)
else:
cart.update(product, product.price, cur_q + 1)
return redirect(reverse('shop:cart'))
def remove_from_cart(request):
try:
sel_prod_id = request.POST['prod_id']
except:
return render(request, 'cart.html', {
'err_msg': 'Product deleting error.'
})
product = get_object_or_404(Product, pk=sel_prod_id)
cart = Cart(request)
cur_q = cart.get_current_quantity(product)
if cur_q > 1:
cart.update(product, product.price, cur_q - 1)
else:
cart.remove(product)
if cart.is_empty():
return redirect(reverse('shop:index'))
return redirect(reverse('shop:cart'))
