pbp_com/apps/store_admin/views.py

""" Admin 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 django.contrib.admin.views.decorators import staff_member_required
from django.contrib.auth.decorators import permission_required

from accounts.utils import get_subscriptions

from catalog.models import (Item, Book, Map, Movie, Person, Subscription,
                            ZipCode)
from store_order.models import Order, OrderItem
from catalog.forms import (MovieForm, MapForm, PersonForm, BookForm,
                           SubscriptionForm, PostalCodeForm)


@staff_member_required
@permission_required('pbp_core.store_admin_login')
def store_admin_home(request):
    """ Store Admin Home page """
    #TODO: security, etc
    return render_to_response("pbp_store/store_admin/home.html",
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def add_edit_subscription(request, sku=None,
                          template='pbp_store/store_admin/add_edit.html'):
    """ Add catalogs form """
    return add_edit(request, Subscription, SubscriptionForm, "Subscription",
                    sku=sku, template=template)

@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def add_edit_book(request, sku=None,
                  template='pbp_store/store_admin/add_edit.html'):
    """ Add / edit book """
    return add_edit(request, Book, BookForm, "Book", sku=sku,
                    template=template)


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def add_edit_map(request, sku=None, template='pbp_store/store_admin/add_edit.html'):
    """ Add / edit map """
    return add_edit(request, Map, MapForm, "Map", sku=sku, template=template)


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def add_edit_movie(request, sku=None,
                   template='pbp_store/store_admin/add_edit.html'):
    """ Add / edit movie """
    return add_edit(request, Movie, MovieForm, "Movie", sku=sku, 
                    template=template)


def add_edit(request, model, modelform, title, sku=None, template=None):
    """ Add edit item """
    item = None
    edit_mode = False
    if sku:
        item = get_object_or_404(model, sku=sku)
        edit_mode = True

    form = modelform(request.POST or None, request.FILES or None,
                     instance=item)
    if form.is_valid():
        item = form.save(commit=False)
        if not edit_mode:
            item.created_by = request.user
        item.last_updated_by = request.user
        item.save()
        if item.is_book:
            form.save_m2m()
        return HttpResponseRedirect(reverse('catalog_list'))

    cancel_url = reverse('catalog_list')

    return render_to_response(template, {'form': form, 'edit_mode': edit_mode,
                              'item_type': title, 'cancel_url': cancel_url },
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def delete_item(request, sku, template='pbp_store/store_admin/delete_confirm.html'):
    """ Add movie """
    item = get_object_or_404(Item, sku=sku)
    if request.method == 'POST':
        item.deleted = True
        item.last_updated_by = request.user
        item.save()
        return HttpResponseRedirect(reverse('catalog_list'))

    return render_to_response(template, {'item': item},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_view')
def list_catalog(request, category='all',
                 template='pbp_store/store_admin/catalog_list.html'):
    """ Simple list of all the catalogs """

    config = {'movie': Movie,
              'book': Book,
              'map': Map,
              'subscription': Subscription,
              'all': Item}

    model = config.get(category, Item)
    # sort by title or price, asc or desc default to title asc
    catalog = model.objects.active()

    return render_to_response(template, {'catalog': catalog,
                                         'category': category},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_admin_orders')
def list_orders(request, template='pbp_store/store_admin/order_list.html'):
    """ Simple list of all the orders """

    config = {'init': Order.INIT,
              'submitted': Order.SUBMITTED,
              'processed': Order.PROCESSED,
              'shipped': Order.SHIPPED,
              'cancelled': Order.CANCELLED}

    status = request.GET.get('status', 'processed')
    if status == 'all':
        orders = Order.objects.all()
    else:
        status_filter = config.get(status)
        orders = Order.objects.filter(status=status_filter)

    return render_to_response(template, {'orders': orders, 'status': status},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_admin_orders')
def view_order_detail(request, order_number,
                      template='pbp_store/store_admin/order.html'):
    """ View order details """
    order = get_object_or_404(Order, pk=order_number)

    return render_to_response(template, {'order': order},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_admin_orders')
def list_subscriptions(
        request, template='pbp_store/store_admin/subscription_list.html'):
    """ Simple list of all the orders """

    config = {'init': Order.INIT,
              'submitted': Order.SUBMITTED,
              'processed': Order.PROCESSED,
              'shipped': Order.SHIPPED,
              'cancelled': Order.CANCELLED}

    status = request.GET.get('status', 'processed')
    if status == 'all':
        order_items = OrderItem.objects.filter(
            product__subclass_type__iexact=Subscription.__name__.lower())
    else:
        status_filter = config.get(status)
        order_items = OrderItem.objects.filter(
            product__subclass_type__iexact=Subscription.__name__.lower()
            ).filter(order__status=status_filter)

    return render_to_response(template, {'order_items': order_items,
                              'status': status},
                              context_instance=RequestContext(request))

@staff_member_required
@permission_required('pbp_core.store_admin_orders')
def subscription_detail(
        request, order_number, order_item=None,
        template='pbp_store/store_admin/subscription_detail.html'):
    """ Simple list of all the orders """

    order = get_object_or_404(Order, pk=order_number)

    if order_number:
        order_items = get_subscriptions(order=order)

    return render_to_response(
        template, {'order_items': order_items, 'order': order},
        context_instance=RequestContext(request))



@staff_member_required
@permission_required('pbp_core.store_catalog_view')
def list_postal_codes(request, template='pbp_store/store_admin/postal_code_list.html'):
    """ list the postal codes """

    postal_codes = ZipCode.objects.active()

    return render_to_response(template, {'postal_codes': postal_codes},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def add_edit_postal_code(request, postal_code=None,
                         template='pbp_store/store_admin/add_edit.html'):
    """ Add edit postal_code """
    item = None
    edit_mode = False
    if postal_code:
        item = get_object_or_404(ZipCode, zipcode=postal_code)
        edit_mode = True

    form = PostalCodeForm(request.POST or None, instance=item)
    if form.is_valid():
        item = form.save(commit=False)
        if not edit_mode:
            item.created_by = request.user
        item.last_updated_by = request.user
        item.save()
        return HttpResponseRedirect(reverse('postal_code_list'))

    cancel_url = reverse('postal_code_list')

    return render_to_response(template, {'form': form, 'edit_mode': edit_mode,
        'item_type': 'Postal Code', 'cancel_url': cancel_url},
        context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def delete_postal_code(request, postal_code,
                       template='pbp_store/store_admin/delete_confirm.html'):
    """ Add movie """
    item = get_object_or_404(ZipCode, zipcode=postal_code)
    if request.method == 'POST':
        item.deleted = True
        item.last_updated_by = request.user
        item.save()
        return HttpResponseRedirect(reverse('postal_code_list'))

    return render_to_response(template, {'item': item},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_view')
def list_persons(request, template='pbp_store/store_admin/person_list.html'):
    """ list the postal codes """

    persons = Person.objects.active()

    return render_to_response(template, {'persons': persons},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def add_edit_person(request, person_slug=None,
                    template='pbp_store/store_admin/add_edit.html'):
    """ Add edit Author """
    item = None
    edit_mode = False
    if person_slug:
        item = get_object_or_404(Person, slug=person_slug)
        edit_mode = True

    form = PersonForm(request.POST or None, request.FILES or None,
                      instance=item)
    if form.is_valid():
        item = form.save(commit=False)
        if not edit_mode:
            item.created_by = request.user
        item.last_updated_by = request.user
        item.save()
        return HttpResponseRedirect(reverse('person_list'))

    cancel_url = reverse('person_list')

    return render_to_response(template, {'form': form, 'edit_mode': edit_mode,
                              'item_type': 'Person', 'cancel_url': cancel_url},
                              context_instance=RequestContext(request))


@staff_member_required
@permission_required('pbp_core.store_catalog_edit')
def delete_person(request, person_slug,
                  template='pbp_store/store_admin/delete_confirm.html'):
    """ Add movie """
    item = get_object_or_404(Person, slug=person_slug)
    if request.method == 'POST':
        item.deleted = True
        item.last_updated_by = request.user
        item.save()
        return HttpResponseRedirect(reverse('person_list'))

    return render_to_response(template, {'item': item},
                              context_instance=RequestContext(request))