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


class Demographics(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'study', 'education', 'concentrated', 'understood']

    def before_next_page(self):
        self.player.set_payoffs()


class Results(Page):

    def vars_for_template(self):
        #payoff_beliefs = self.player.payoffBeliefA+self.player.payoffBeliefB
        donations_average = (self.player.donations1 + self.player.donations2 + self.player.donations3)/3
        average_section1 = (self.player.payoff1 + self.player.payoff2 + self.player.payoff3)/3
        payoff_section1 = self.player.totalPayoffPoints
        payoff_section1_euro = payoff_section1*Constants.exchange_rate
        payoff_section2 = (self.player.payoff_svo + self.player.payoff_risk)/100
        return dict(#payoff_beliefs=payoff_beliefs,
                    donations_average=round(donations_average,1),
                    average_section1 = round(average_section1,1), payoff_section1=round(payoff_section1,1),
                    payoff_section1_euro=round(payoff_section1_euro,1), payoff_section2=payoff_section2)


page_sequence = [Demographics,
                 Results]