from otree.api import Currency as c, currency_range

from ._builtin import Page, WaitPage
from .models import Constants
import pandas as pd
bookkeeper_odds_df = pd.read_excel('prescreening_UK/bookkeeper_odds.xlsx')

class MatchBeliefs_a(Page):
    form_model = 'player'
    form_fields = ['match_1_win', 'match_1_draw', 'match_1_lose', 'match_1_sum', 'MatchBeliefs_a_time']

    def vars_for_template(self):
        return dict(
            match_num=self.player.participant.vars['task_rounds']['A'],
            home_team1=bookkeeper_odds_df['match1_home_team'].iloc[0],
            away_team1=bookkeeper_odds_df['match1_away_team'].iloc[0],
            home_team1_abbv=bookkeeper_odds_df['match1_home_team_abbv'].iloc[0],
            away_team1_abbv=bookkeeper_odds_df['match1_away_team_abbv'].iloc[0],
        )

    def error_message(self, values):
        for field_name in ['match_1_win', 'match_1_draw', 'match_1_lose']:
            if values[field_name] > 98 or values[field_name] == 0:
                return 'Please do not select 100 or 0 as options'

    def is_displayed(self):
        return self.player.round_number == self.player.participant.vars['task_rounds']['A']

    def before_next_page(self):
        self.player.participant.vars['match_1_win'] = self.player.match_1_win
        self.player.participant.vars['match_1_draw'] = self.player.match_1_draw
        self.player.participant.vars['match_1_lose'] = self.player.match_1_lose

class MatchBeliefs_b(Page):
    form_model = 'player'
    form_fields = ['match_2_win', 'match_2_draw', 'match_2_lose', 'match_2_sum', 'MatchBeliefs_b_time']

    def vars_for_template(self):
        return dict(
            match_num=self.player.participant.vars['task_rounds']['B'],
            home_team2=bookkeeper_odds_df['match2_home_team'].iloc[0],
            away_team2=bookkeeper_odds_df['match2_away_team'].iloc[0],
            home_team2_abbv=bookkeeper_odds_df['match2_home_team_abbv'].iloc[0],
            away_team2_abbv=bookkeeper_odds_df['match2_away_team_abbv'].iloc[0],
        )
    def error_message(self, values):
        for field_name in ['match_2_win', 'match_2_draw', 'match_2_lose']:
            if values[field_name] > 98 or values[field_name] == 0:
                return 'Please do not select 100 or 0 as options'

    def is_displayed(self):
        return self.player.round_number == self.player.participant.vars['task_rounds']['B']

    def before_next_page(self):
        self.player.participant.vars['match_2_win'] = self.player.match_2_win
        self.player.participant.vars['match_2_draw'] = self.player.match_2_draw
        self.player.participant.vars['match_2_lose'] = self.player.match_2_lose

class MatchBeliefs_c(Page):
    form_model = 'player'
    form_fields = ['match_3_win', 'match_3_draw', 'match_3_lose', 'match_3_sum', 'MatchBeliefs_c_time']

    def vars_for_template(self):
        return dict(
            match_num=self.player.participant.vars['task_rounds']['C'],
            home_team3=bookkeeper_odds_df['match3_home_team'].iloc[0],
            away_team3=bookkeeper_odds_df['match3_away_team'].iloc[0],
            home_team3_abbv=bookkeeper_odds_df['match3_home_team_abbv'].iloc[0],
            away_team3_abbv=bookkeeper_odds_df['match3_away_team_abbv'].iloc[0]
        )

    def error_message(self, values):
        for field_name in ['match_3_win', 'match_3_draw', 'match_3_lose']:
            if values[field_name] > 98 or values[field_name] == 0:
                return 'Please do not select 100 or 0 as options'

    def is_displayed(self):
        return self.player.round_number == self.player.participant.vars['task_rounds']['C']

    def before_next_page(self):
        self.player.participant.vars['match_3_win'] = self.player.match_3_win
        self.player.participant.vars['match_3_draw'] = self.player.match_3_draw
        self.player.participant.vars['match_3_lose'] = self.player.match_3_lose

class MatchBeliefs_d(Page):
    form_model = 'player'
    form_fields = ['match_4_win', 'match_4_draw', 'match_4_lose', 'match_4_sum', 'MatchBeliefs_d_time']

    def vars_for_template(self):
        return dict(
            match_num=self.player.participant.vars['task_rounds']['D'],
            home_team4=bookkeeper_odds_df['match4_home_team'].iloc[0],
            away_team4=bookkeeper_odds_df['match4_away_team'].iloc[0],
            home_team4_abbv=bookkeeper_odds_df['match4_home_team_abbv'].iloc[0],
            away_team4_abbv=bookkeeper_odds_df['match4_away_team_abbv'].iloc[0]
        )

    def error_message(self, values):
        for field_name in ['match_4_win', 'match_4_draw', 'match_4_lose']:
            if values[field_name] > 98 or values[field_name] == 0:
                return 'Please do not select 100 or 0 as options'

    def is_displayed(self):
        return self.player.round_number == self.player.participant.vars['task_rounds']['D']

    def before_next_page(self):
        self.player.participant.vars['match_4_win'] = self.player.match_4_win
        self.player.participant.vars['match_4_draw'] = self.player.match_4_draw
        self.player.participant.vars['match_4_lose'] = self.player.match_4_lose

page_sequence = [MatchBeliefs_a,
                 MatchBeliefs_b,
                 MatchBeliefs_c,
                 MatchBeliefs_d,
                 ]