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)