from otree.api import Currency as c, currency_range

from ._builtin import Page, WaitPage
from .models import Constants

class NBO_choice(Page):
    form_model = 'player'
    form_fields = ['nbo_choice']
    
    def before_next_page(self):
        if self.player.nbo_choice:
            self.player.set_value()
            self.player.set_payoff()

class o10a40(Page):
    form_model = 'player'
    form_fields = ['option_choose']
    def is_displayed(self):
        return self.player.nbo_choice == False
    def vars_for_template(self):
        return dict(
            image_path_1='no_choice/option_images/d_{}_o_1.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_2='no_choice/option_images/d_{}_o_2.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_3='no_choice/option_images/d_{}_o_3.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_4='no_choice/option_images/d_{}_o_4.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_5='no_choice/option_images/d_{}_o_5.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_6='no_choice/option_images/d_{}_o_6.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_7='no_choice/option_images/d_{}_o_7.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_8='no_choice/option_images/d_{}_o_8.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_9='no_choice/option_images/d_{}_o_9.png'.format(self.participant.vars['order'][self.round_number-1]),
            image_path_10='no_choice/option_images/d_{}_o_10.png'.format(self.participant.vars['order'][self.round_number-1]),
            problem = self.participant.vars['order'][self.round_number-1]
        #Now need to add vars for option labels (?) and values for each option 
        )
    
    def before_next_page(self):
            self.player.set_value()
            self.player.set_payoff()
    
    
#class ResultsWaitPage(WaitPage):

            
class Results(Page):
    def vars_for_template(self):
        me = self.player
        return dict(
            my_decision=me.option_choose,
            my_value=me.value
        )

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

        return dict(
            my_payoff= self.player.in_round(self.player.participant.vars['pay_round']).payoff,
            my_pay_round = self.player.participant.vars['pay_round'],
            my_nbo_choice = self.player.in_round(self.player.participant.vars['pay_round']).nbo_choice
        )


page_sequence = [
    NBO_choice,
    o10a40,
    Results,
    Final_Results
]