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]