from django.contrib import admin
from leaflet.admin import LeafletGeoAdmin

from .models import CaseStudy


class CaseStudyAdmin(LeafletGeoAdmin):
    list_display = ('id', 'date_created', 'entry_name', 'approved')
    actions = ['approve', 'unapprove']

    def approve(self, request, queryset):
        updated = queryset.update(approved=True)
        if updated == 1:
            message_bit = "1 case study was"
        else:
            message_bit = "{0} case studies were".format(updated)
        self.message_user(request, "{0} successfully approved".format(
            message_bit
        ))
    approve.short_description = "Approve selected case studies"

    def unapprove(self, request, queryset):
        updated = queryset.update(approved=False)
        if updated == 1:
            message_bit = "1 case study was"
        else:
            message_bit = "{0} case studies were".format(updated)
        self.message_user(request, "{0} successfully un-approved".format(
            message_bit
        ))
    unapprove.short_description = "Un-approve selected case studies"

admin.site.register(CaseStudy, CaseStudyAdmin)