from django.http import HttpResponseRedirect
from django.template import RequestContext
from django.shortcuts import render_to_response
from django import forms
from django.db import transaction
from django.contrib import admin
from django.contrib.contenttypes.generic import GenericStackedInline
from .models import Chain, Store, Address, Brand, ClaimRequest, Link, LinkType, County, Country


class LinkInlineAdmin(GenericStackedInline):
    model = Link
    ct_field = 'object_type'


class ChainAdmin(admin.ModelAdmin):
    list_filter = ['active']
    list_display = ['name']
    prepopulated_fields = {"slug": ("name",)}
    search_fields = ['name']
    inlines = [
        LinkInlineAdmin,
    ]

class StoreAdmin(admin.ModelAdmin):
    list_filter = ['chain', 'active']
    list_display = ['name', 'store_type', 'active']
    prepopulated_fields = {"slug": ("name",)}
    search_fields = ['name']
    inlines = [
        LinkInlineAdmin,
    ]
    actions = ['add_brand', 'set_chain']

    class AddBrandForm(forms.Form):
        _selected_action = forms.CharField(widget=forms.MultipleHiddenInput)
        brand = forms.ModelChoiceField(Brand.objects)

    def add_brand(self, request, queryset):
        form = None
        if 'apply' in request.POST:
            form = self.AddBrandForm(request.POST)
            if form.is_valid():
                count = queryset.count()
                brand = form.cleaned_data['brand']
                with transaction.commit_on_success():
                    for store in queryset:
                        store.brands.add(brand)
                if count > 1:
                    plural = 's'
                else:
                    plural = ''
                self.message_user(request, "Successfully added %s to %d store%s." % (brand, count, plural))
                return HttpResponseRedirect(request.get_full_path())
        if not form:
            form = self.AddBrandForm(initial={'_selected_action': queryset.values_list('id', flat=True)})
        return render_to_response('admin/add_brand.html', {'stores': queryset, 'brand_form': form }, RequestContext(request))

    add_brand.short_description = "Add brand to the selected stores"

    class SetChainForm(forms.Form):
        _selected_action = forms.CharField(widget=forms.MultipleHiddenInput)
        chain = forms.ModelChoiceField(Chain.objects)

    def set_chain(self, request, queryset):
        form = None
        if 'apply' in request.POST:
            form = self.SetChainForm(request.POST)
            if form.is_valid():
                count = queryset.count()
                chain = form.cleaned_data['chain']
                with transaction.commit_on_success():
                    for store in queryset:
                        store.chain = chain
                        store.save()
                if count > 1:
                    plural = 's'
                else:
                    plural = ''
                self.message_user(request, "Successfully set %d store%s to %s." % (count, plural, chain))
                return HttpResponseRedirect(request.get_full_path())
        if not form:
            form = self.SetChainForm(initial={'_selected_action': queryset.values_list('id', flat=True)})
        return render_to_response('admin/set_chain.html', {'stores': queryset, 'chain_form': form }, RequestContext(request))

    set_chain.short_description = "Set the selected store's chain"


class ClaimAdmin(admin.ModelAdmin):
    list_filter = ['status']
    list_display = ['generic_obj', 'user', 'status', 'note']
    actions = ['approve_request']


    def approve_request(self, request, queryset):
        qs = queryset.filter(status=ClaimRequest.CLAIM_STATUS_PENDING)
        for obj in qs:
            obj.status = ClaimRequest.CLAIM_STATUS_APPROVED
            target = obj.generic_obj
            target.editor = obj.user
            target.save()
            obj.save()
        if qs.count() == 1:
            message_bit = "1 request was"
        else:
            message_bit = "%s requests were" % qs.count()
        self.message_user(request, "%s successfully approved." % message_bit)

    approve_request.short_description = 'Approve selected requests.'


admin.site.register(Chain, ChainAdmin)
admin.site.register(Store, StoreAdmin)
admin.site.register(Address, admin.ModelAdmin)
admin.site.register(Brand, admin.ModelAdmin)
admin.site.register(ClaimRequest, ClaimAdmin)
admin.site.register(LinkType, admin.ModelAdmin)
admin.site.register(County, admin.ModelAdmin)
admin.site.register(Country, admin.ModelAdmin)