add options for the admin page

django_code/apps/random_simpsons/admin.py

@@ -1,6 +1,8 @@
 from django.contrib import admin,messages
 from django.utils.translation import ngettext
+from prompt_toolkit.shortcuts import print_container
 from .models import SimpsonsData,simpsons_data_object
+from .utils import mark_episode_watched 
 from import_export import resources
 from import_export.admin import ImportExportModelAdmin
 # Register your models here.
@@ -10,6 +12,13 @@ DB = 'simpsons_db'
 @admin.action(description="Mark episode as seen")
 def make_seen(modeladmin, request, queryset):
     updated = queryset.update(seen="True")
+    
+    for query in queryset.values():
+        print(query)
+        season_number = query.get('season')
+        episode_number = query.get('episode')
+        mark_episode_watched(season_number,episode_number,"watched")
+
     modeladmin.message_user(
         request,
         ngettext(
@@ -24,6 +33,12 @@ def make_seen(modeladmin, request, queryset):
 @admin.action(description="Mark episode as not seen")
 def make_unseen(modeladmin, request, queryset):
     updated = queryset.update(seen="False")
+    for query in queryset.values():
+        print(query)
+        season_number = query.get('season')
+        episode_number = query.get('episode')
+        mark_episode_watched(season_number,episode_number,"unwatched")
+        
     modeladmin.message_user(
         request,
         ngettext(