pbp_com/apps/cart/views.py

""" cart.views """
from django.core.urlresolvers import reverse
from django.http import HttpResponseRedirect
from django.shortcuts import render_to_response, get_object_or_404
from django.template import RequestContext

from catalog.models import Item
from .models import CartItem

from .cart import add_to_cart, get_cart, remove_from_cart, update_cart


def add_item_to_cart(request, sku):
    """Add item to cart """
    item = get_object_or_404(Item, sku=sku)
    add_to_cart(request, item, quantity=1)
    return HttpResponseRedirect(reverse('shopping_cart'))


def remove_item_from_cart(request, cart_token):
    """Add item to cart """
    remove_from_cart(request, cart_token)
    return HttpResponseRedirect(reverse('shopping_cart'))


def __is_quantity_valid(qty):
    """ validate the quantity value"""
    try:
        qty_num = int(qty)
    except ValueError:
        return False
    if qty_num < 0:
        return False
    if qty_num > 1000:
        return False
    return True


def update_item_in_cart(request):
    """ Update item quantity """
    if request.method == 'POST':
        postdata = request.POST.copy()
        item_token = postdata.get('item', None)
        qty = postdata.get('quantity', 0)
        item = get_object_or_404(CartItem, cart_token=item_token)
        if not __is_quantity_valid(qty):
            return shopping_cart(request, error_message='Quantity invalid')
        update_cart(request, item, qty)

    return HttpResponseRedirect(reverse('shopping_cart'))


def shopping_cart(request, messages=None, error_message=None):
    """ Show the shopping cart """
    cart = get_cart(request)
    return render_to_response("cart/shopping_cart.html",
                              {'cart': cart, 'messages': messages,
                               'error_message': error_message},
                              context_instance=RequestContext(request))