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()

simpsons_id = 12

def mark_episode_watched(season,episode,action):
    
    print("season_number: "+str(season))
    print("episode_number: "+str(episode))
    
    episode_number = "S"+str(season)+"E"+str(episode)
    
    episode_id = betaserie.identify_episode(simpsons_id,episode_number)
    
    if action == "watched":
        betaserie.mark_episodes_watched(episode_id)
    elif action == "unwatched":
        betaserie.unmark_episodes_watched(episode_id)
    else:
        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")