from otree.api import Currency as c, currency_range
from . import models
from ._builtin import Page, WaitPage
from .models import Constants
import random
import random
import pandas as pd
import itertools

import numpy as np

from sklearn.utils import shuffle


class MyPage(Page):
    pass


class ResultsWaitPage(WaitPage):

    def after_all_players_arrive(self):
        pass

class ENG_choice_page(Page):
    form_model = models.Player
    form_fields = ['risk_choice']
    def before_next_page(self):
        self.player.save_my_choice()
        self.player.subject_risk_payout()
        self.participant.vars['risk_payout'] = self.player.risk_payout


# class Results(Page):
#     def vars_for_template(self):
#         self.player.whole_shebang()
#     #
#     def after_all_players_arrive(self):
#         self.player.whole_shebang()

class Final_page(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds

class Final(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds
    def vars_for_template(self):
        # self.player.round_to_pay()
        self.player.calculate_final_payout()


    def after_all_players_arrive(self):
        # self.player.round_to_pay()
        self.player.calculate_final_payout()


    def before_session_starts(self):
        self.round_list = range(1, self.round_number + 1)
        self.payout_chosen_trial = shuffle(self.round_list)

        self.participant.vars['payout_chosen_trial'] = self.payout_chosen_trial[0]



page_sequence = [
    ENG_choice_page,
    Final
    # Results,
    # Final_page,
]