34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
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)
|