40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from django.contrib import admin,messages
|
|
from django.utils.translation import ngettext
|
|
from .models import SimpsonsData
|
|
# Register your models here.
|
|
|
|
|
|
@admin.action(description="Mark episode as seen")
|
|
def make_seen(modeladmin, request, queryset):
|
|
updated = queryset.update(seen="True")
|
|
modeladmin.message_user(
|
|
request,
|
|
ngettext(
|
|
"%d episode was successfully marked as seen.",
|
|
"%d episodes were successfully marked as seen.",
|
|
updated,
|
|
)
|
|
% updated,
|
|
messages.SUCCESS,
|
|
)
|
|
|
|
@admin.action(description="Mark episode as not seen")
|
|
def make_unseen(modeladmin, request, queryset):
|
|
updated = queryset.update(seen="False")
|
|
modeladmin.message_user(
|
|
request,
|
|
ngettext(
|
|
"%d episode was successfully marked as not seen.",
|
|
"%d episodes were successfully marked as not seen.",
|
|
updated,
|
|
)
|
|
% updated,
|
|
messages.SUCCESS,
|
|
)
|
|
|
|
@admin.register(SimpsonsData)
|
|
class EpisodeAdmin(admin.ModelAdmin):
|
|
list_display = ('season', 'episode', 'url', 'seen')
|
|
list_filter = ['seen']
|
|
fields = ['season', 'episode', 'url', 'seen']
|
|
actions = [make_seen,make_unseen] |