root/trunk/djedna/store/views.py

# (c) Copyright 2008 Thomas Bohmbach, Jr.
# 
# This file is part of DJ Edna.
# 
# DJ Edna is free software: you can redistribute it and/or modify it under the
# terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
# 
# DJ Edna is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
# or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
# 
# You should have received a copy of the GNU General Public License along with
# DJ Edna.  If not, see <http://www.gnu.org/licenses/>.

import decimal
import logging as log

from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.contrib.auth.models import User
from django.core.urlresolvers import reverse
from django.http import HttpResponse, HttpResponseRedirect, HttpResponseNotFound, HttpResponseForbidden, Http404, HttpResponseServerError
from django.template import loader, Context, RequestContext
from django.shortcuts import render_to_response, get_object_or_404
from django.views.generic.list_detail import object_list, object_detail

from forms import create_price_form, create_price_forms
from models import UserAccessPermitProduct


def product_list(request, content_object=None):
    cart = None
    if request.user and request.user.is_authenticated():
        cart = request.user.get_profile().cart
    if content_object:
        products = UserAccessPermitProduct.objects.get_products(user, content_object)
    else:
        products = UserAccessPermitProduct.objects.all()
    query_set = UserAccessPermitProduct.objects.filter(id__in=[product.id for product in products if product.is_valid()])
    return object_list(request,
                       query_set,
                       template_name='store/product_list.html',
                       template_object_name='product',
                       extra_context={'cart' : cart})

def product_detail(request, product_id):
    cart = None
    if request.user and request.user.is_authenticated():
        cart = request.user.get_profile().cart
    query_set = UserAccessPermitProduct.objects.all()
    return object_detail(request,
                         query_set,
                         object_id=product_id,
                         template_name='store/product_detail.html',
                         template_object_name='product',
                         extra_context={'cart' : cart})

#TODO: Get the cart from a utility method that checks the session if the user isn't logged in
#TODO: When a user registers, their session cart should be assigned to that user
#TODO: When a user logs in, if they have a session cart it should replace (merge with) their user cart
@login_required
def buy_product_form(request, product_id, next_page=None):
    product = get_object_or_404(UserAccessPermitProduct, pk=product_id)
    cart = request.user.get_profile().cart
    next_page = next_page or request.REQUEST.get('next_page', '') or product.content_object.get_absolute_url()
    if request.method == 'POST':
        form = create_price_form(product, data=request.POST)
        if form.is_valid():
            price = form.final_price
            cart.add(product, price)
            return HttpResponseRedirect(next_page)
    else:
        form = create_price_form(product)
    return render_to_response('store/buy_product_form.html',
                              {'next_page' : next_page, 'form' : form, 'product' : product, 'cart' : cart},
                              context_instance=RequestContext(request))

@login_required
def buy_products_form(request, products=[], next_page=None):
    cart = request.user.get_profile().cart
    if not products:
        all_products = UserAccessPermitProduct.objects.all()
        products = UserAccessPermitProduct.objects.filter(
            id__in=[product.id for product in all_products if (product.is_valid() and not cart.contains_product(product))]
        )
    next_page = next_page or request.REQUEST.get('next_page', request.get_full_path())
    product_forms = []
    if request.method == 'POST':
        #We don't want to validate all the forms yet, so just create blank ones
        product_forms, selected_form = create_price_forms(products)
        for product, form in product_forms:
            #Did the user hit the AddToCart button for this form and is the form valid?
            if request.POST.get('%s-add_to_cart' % form.prefix, None):
                #Now that we know which product was selected, create the forms, only validating that one
                product_forms, selected_form = create_price_forms(products, data=request.POST, selected=product)
                if selected_form.is_valid():
                    cart.add(product, selected_form.final_price)
                    return HttpResponseRedirect(next_page)
    else:
        product_forms, selected_form = create_price_forms(products)
    return render_to_response('store/buy_products_form.html',
                              {'next_page' : next_page, 'product_forms' : product_forms, 'cart' : cart},
                              context_instance=RequestContext(request))
    

@login_required
def buy_user_access_permit(request, product_id, price):
    product = get_object_or_404(UserAccessPermitProduct, pk=product_id)
    price = decimal.Decimal(price)
    cart = request.user.get_profile().cart
    cart.add(product, price)
    return render_to_response('store/cart_add.html',
                              {'cart' : cart,
                               'product' : product},
                              context_instance=RequestContext(request))

@login_required
def remove_from_cart(request, next_page=None):
    log.info("next_page=%s" % next_page)
    next_page = next_page or request.REQUEST.get('next_page', request.get_full_path())
    cart = request.user.get_profile().cart
    for cart_item in cart.items.all():
        if request.POST.get('%s-remove_from_cart' % cart_item.id, None):
            cart.remove(cart_item.product)
    return HttpResponseRedirect(next_page)

@login_required
def cart_clear(request):
    cart = request.user.get_profile().cart
    cart.clear()

@login_required
def checkout(request):
    pass

@login_required
def review(request):
    pass

@login_required
def purchase(request):
    cart = request.user.get_profile().cart
    cart.purchase()