from otree.api import Currency as c, currency_range

from ._builtin import Page, WaitPage
from .models import Constants
import pandas as pd
from math import sqrt

class Introduction(Page):
    form_model = 'player'
    form_fields = ['agree_survey', 'Introduction_time']

class EarlyDropout(Page):
    form_model = 'player'
    form_fields = []

    def is_displayed(self):
        return self.player.agree_survey == 0

class fav_club(Page):
    form_model = 'player'
    form_fields = ['fav_club', 'fav_club_time']

class fav_club_supp(Page):
    form_model = 'player'
    form_fields = ['fav_club_supp', 'fav_club_supp_time']

    def is_displayed(self):
        return self.player.fav_club != "None"

class fav_club_2nd(Page):
    form_model = 'player'
    form_fields = ['fav_club_2nd', 'fav_club_2nd_time']

    def is_displayed(self):
        return self.player.fav_club != "None"

class fav_club_2nd_supp(Page):
    form_model = 'player'
    form_fields = ['fav_club_2nd_supp', 'fav_club_2nd_supp_time']

    def is_displayed(self):
        return self.player.fav_club != "None" and self.player.fav_club_2nd != "None"

class least_fav_club(Page):
    form_model = 'player'
    form_fields = ['least_fav_club', 'least_fav_club_time']

class least_fav_club_dislike(Page):
    form_model = 'player'
    form_fields = ['least_fav_club_dislike', 'least_fav_club_dislike_time']

    def is_displayed(self):
        return self.player.least_fav_club != "None"

class least_fav_club_2nd(Page):
    form_model = 'player'
    form_fields = ['least_fav_club_2nd', 'least_fav_club_2nd_time']

    def is_displayed(self):
        return self.player.least_fav_club != "None"

class least_fav_club_2nd_dislike(Page):
    form_model = 'player'
    form_fields = ['least_fav_club_2nd_dislike', 'least_fav_club_2nd_dislike_time']
    def is_displayed(self):
        return self.player.least_fav_club != "None" and self.player.least_fav_club_2nd != "None"

class Introduction_ranking(Page):
    form_model = 'player'
    form_fields = ['Introduction_ranking_time']

class chelsea_rank(Page):
    form_model = 'player'
    form_fields = ['chelsea_rank', 'chelsea_certain', 'chelsea_rank_time']

    def vars_for_template(self):

        return dict(
            team = "Chelsea"
        )

class arsenal_rank(Page):
    form_model = 'player'
    form_fields = ['arsenal_rank', 'arsenal_certain', 'arsenal_rank_time']

    def vars_for_template(self):

        return dict(
            team = "FC Arsenal"
        )

class liverpool_rank(Page):
    form_model = 'player'
    form_fields = ['liverpool_rank', 'liverpool_certain', 'liverpool_rank_time']

    def vars_for_template(self):

        return dict(
            team = "FC Liverpool"
        )

class manu_rank(Page):
    form_model = 'player'
    form_fields = ['manu_rank', 'manu_certain', 'manu_rank_time']

    def vars_for_template(self):

        return dict(
            team = "Manchester United"
        )

class mancity_rank(Page):
    form_model = 'player'
    form_fields = ['mancity_rank', 'mancity_certain', 'mancity_rank_time']

    def vars_for_template(self):

        return dict(
            team = "Manchester City"
        )

class spurs_rank(Page):
    form_model = 'player'
    form_fields = ['spurs_rank', 'spurs_certain', 'spurs_rank_time']

    def vars_for_template(self):

        return dict(
            team = "Tottenham"
        )

class newcastle_rank(Page):
    form_model = 'player'
    form_fields = ['newcastle_rank', 'newcastle_certain', 'newcastle_rank_time']

    def vars_for_template(self):

        return dict(
            team = "Newcastle United"
        )

class ThankYou_End(Page):
    form_model = 'player'
    form_fields = []

    def js_vars(self):
        link = "https://app.prolific.co/submissions/complete?cc=CTWAXFWI"

        return dict(url=link)

page_sequence = [Introduction,
                 EarlyDropout,
                 fav_club,
                 fav_club_supp,
                 fav_club_2nd,
                 fav_club_2nd_supp,
                 least_fav_club,
                 least_fav_club_dislike,
                 least_fav_club_2nd,
                 least_fav_club_2nd_dislike,
                 Introduction_ranking,
                 chelsea_rank,
                 arsenal_rank,
                 liverpool_rank,
                 manu_rank,
                 mancity_rank,
                 spurs_rank,
                 newcastle_rank,
                 ThankYou_End,
                 ThankYou_End
                 ]