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 PotentialBonus(Page):
    form_model = 'player'
    form_fields = ['PotentialBonus_time']

    def vars_for_template(self):
        if self.player.payout_matchnum == 1:
            home_team = bookkeeper_odds_df['match1_home_team'].iloc[0]
            away_team = bookkeeper_odds_df['match1_away_team'].iloc[0]
            home_team_winodds = self.player.participant.vars['matchbet_1_winodds']
            home_team_loseodds = self.player.participant.vars['matchbet_1_loseodds']
            draw_odds = bookkeeper_odds_df['match1_draw_odds'].iloc[0]
            win_bet = self.player.participant.vars['matchbet_1_change_win']
            lose_bet = self.player.participant.vars['matchbet_1_change_lose']
            draw_bet = self.player.participant.vars['matchbet_1_change_draw']
            win_payoff = round(win_bet * home_team_winodds, 2)
            lose_payoff = round(lose_bet * home_team_loseodds, 2)
            draw_payoff = round(draw_bet * draw_odds, 2)
        elif self.player.payout_matchnum == 2:
            home_team = bookkeeper_odds_df['match2_home_team'].iloc[0]
            away_team = bookkeeper_odds_df['match2_away_team'].iloc[0]
            home_team_winodds = self.player.participant.vars['matchbet_2_winodds']
            home_team_loseodds = self.player.participant.vars['matchbet_2_loseodds']
            draw_odds = bookkeeper_odds_df['match2_draw_odds'].iloc[0]
            win_bet = self.player.participant.vars['matchbet_2_change_win']
            lose_bet = self.player.participant.vars['matchbet_2_change_lose']
            draw_bet = self.player.participant.vars['matchbet_2_change_draw']
            win_payoff = round(win_bet * home_team_winodds, 2)
            lose_payoff = round(lose_bet * home_team_loseodds, 2)
            draw_payoff = round(draw_bet * draw_odds, 2)
        elif self.player.payout_matchnum == 3:
            home_team = bookkeeper_odds_df['match3_home_team'].iloc[0]
            away_team = bookkeeper_odds_df['match3_away_team'].iloc[0]
            home_team_winodds = self.player.participant.vars['matchbet_3_winodds']
            home_team_loseodds = self.player.participant.vars['matchbet_3_loseodds']
            draw_odds = bookkeeper_odds_df['match3_draw_odds'].iloc[0]
            win_bet = self.player.participant.vars['matchbet_3_change_win']
            lose_bet = self.player.participant.vars['matchbet_3_change_lose']
            draw_bet = self.player.participant.vars['matchbet_3_change_draw']
            win_payoff = round(win_bet * home_team_winodds, 2)
            lose_payoff = round(lose_bet * home_team_loseodds, 2)
            draw_payoff = round(draw_bet * draw_odds, 2)
        elif self.player.payout_matchnum == 4:
            home_team = bookkeeper_odds_df['match4_home_team'].iloc[0]
            away_team = bookkeeper_odds_df['match4_away_team'].iloc[0]
            home_team_winodds = self.player.participant.vars['matchbet_4_winodds']
            home_team_loseodds = self.player.participant.vars['matchbet_4_loseodds']
            draw_odds = bookkeeper_odds_df['match4_draw_odds'].iloc[0]
            win_bet = self.player.participant.vars['matchbet_4_change_win']
            lose_bet = self.player.participant.vars['matchbet_4_change_lose']
            draw_bet = self.player.participant.vars['matchbet_4_change_draw']
            win_payoff = round(win_bet * home_team_winodds, 2)
            lose_payoff = round(lose_bet * home_team_loseodds, 2)
            draw_payoff = round(draw_bet * draw_odds, 2)

        return dict(
            home_team = home_team,
            away_team = away_team,
            home_team_winodds = home_team_winodds,
            home_team_loseodds = home_team_loseodds,
            draw_odds = draw_odds,
            win_bet = win_bet,
            lose_bet = lose_bet,
            draw_bet = draw_bet,
            win_payoff = win_payoff,
            lose_payoff = lose_payoff,
            draw_payoff = draw_payoff
        )

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

        return dict(url=link)

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

page_sequence = [
                 PotentialBonus,
                 ThankYou
                 ]