from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants



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

    def before_next_page(self):
        if self.player.id_in_group==1:
            self.group.position1=self.player.Choice
        else:
            self.group.position2 = self.player.Choice



class ResultsWaitPage(WaitPage):
    pass

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

    def before_next_page(self):
        self.player.gain1 = Constants.Gain_Matrix_p1[self.group.position1 - 1, self.group.position2 - 1]
        self.player.gain2 = Constants.Gain_Matrix_p2[self.group.position1 - 1, self.group.position2 - 1]
        if self.player.id_in_group==1 and self.player.Payoff_choice==0:
            self.player.gain=self.player.gain1
        elif self.player.id_in_group==2 and self.player.Payoff_choice==0:
            self.player.gain = self.player.gain2
        elif self.player.Payoff_choice==1:
            self.player.gain =Constants.s_payoff

class Results(Page):
    form_model = 'player'
    form_field=[]
    def vars_for_template(self):
        return dict(other_player_choice=self.player.other_player().Choice)


page_sequence = [Decision,
                 ResultsWaitPage,
                 Decision2,
                 ResultsWaitPage,
                 Results]