adding the unseen all option

django_code/apps/random_simpsons/utils.py

@@ -1,5 +1,8 @@
 from apps.betaserie.utils import betaserieUtils
+from django.db.models import Count
 from celery.worker.state import active_requests
+from requests.compat import str
+from .models import simpsons_data_object
 
 betaserie = betaserieUtils()
 
@@ -19,4 +22,16 @@ def mark_episode_watched(season,episode,action):
     elif action == "unwatched":
         betaserie.unmark_episodes_watched(episode_id)
     else:
-        print("Parameter action not in range")
+        print("Parameter action not in range")
+
+
+
+def unwatch_all_episodes():
+    maximum_number_of_seasons = simpsons_data_object.aggregate(unique_season=Count('season', distinct=True))['unique_season']
+    print("maximum_number_of_seasons: "+str(maximum_number_of_seasons))
+    for season in range(1,maximum_number_of_seasons+1):
+        maximum_number_of_episodes = simpsons_data_object.filter(season=season).count()
+        print("maximum_number_of_episodes for "+str(season)+": "+str(maximum_number_of_episodes))
+        for episode in range(1,maximum_number_of_episodes+1):
+            print("S"+str(season)+"E"+str(episode))
+            mark_episode_watched(season,episode,"unwatched")