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,
]