from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants
def vars_for_all_templates(self):
    return {
        'test_BDM_r': self.player.participant.vars['test_BDM_r'],
        'sure_payoff_r': self.player.participant.vars['sure_payoff_r'],
        'test_BDM_payoff': (self.player.participant.vars['test_BDM_r']-1)*3 ,
        'selected_choice':self.player.participant.vars['selected_choice'],
        'lottery_payoff_r':self.player.participant.vars['lottery_payoff_r'],
        'a':(self.player.participant.vars['test_BDM_r']-1)*3+ self.player.participant.vars['lottery_payoff_r'],
        'change':self.player.Change,
        'b0': (self.player.participant.vars['test_BDM_r']-1)*3 ,
        'b1': (self.player.participant.vars['test_BDM_r']-1)*3 +100,



        'test_BDM_r2': self.player.participant.vars['test_BDM_r2'],
        'sure_payoff_r2': self.player.participant.vars['sure_payoff_r2'],
        'test_BDM_payoff2': (self.player.participant.vars['test_BDM_r2']-1)*3 ,
        'selected_choice2':self.player.participant.vars['selected_choice2'],
        'lottery_payoff_r2':self.player.participant.vars['lottery_payoff_r2'],
        'a2':(self.player.participant.vars['test_BDM_r']-1)*3+ self.player.participant.vars['lottery_payoff_r2'],
        'change2':self.player.Change2,
        'lottery_payoff_r_number':self.player.participant.vars['lottery_payoff_r_number']


    }




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

    def before_next_page(self):
        self.player.participant.vars['sure_payoff_r2']= (self.player.Change2-1)*3


        if self.player.Change2>= self.player.participant.vars['test_BDM_r2']:
            self.player.participant.vars['selected_choice2']=0
        else:
            self.player.participant.vars['selected_choice2']=1




class Results_2(Page):
    def vars_for_template(self):
        return {
            'change2': (self.player.Change2 - 1) * 3,
            'change3': self.player.Change2,
    }




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

    def before_next_page(self):
        self.player.participant.vars['sure_payoff_r']= (self.player.Change-1)*3


        if self.player.Change>= self.player.participant.vars['test_BDM_r']:
            self.player.participant.vars['selected_choice']=0
        else:
            self.player.participant.vars['selected_choice']=1



class Results(Page):
    def vars_for_template(self):
        return {
            'change2': (self.player.Change - 1) * 3,
    }

page_sequence = [MyPage, Results]