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


class RiskAversion(Page):
    form_model = 'player'
    form_fields = ['ra1', 'ra2', 'ra3', 'ra4', 'ra5', 'ra6', 'ra7', 'ra8', 'ra9', 'ra10']

    def before_next_page(self):
        self.player.set_payoff()
    pass


class Demographics(Page):
    form_model = 'player'
    form_fields = ['birth_year', 'self_gender', 'high_school_country', 'ethnicity']
    pass


class Studies(Page):
    form_model = 'player'
    form_fields = ['area', 'year_of_study', 'student_in_major', 'type_major', 'desired_major']
    pass


class Finance(Page):
    form_model = 'player'
    form_fields = ['finance1', 'finance2', 'finance3', 'finance4', 'finance5']
    pass


class Final(Page):
    def vars_for_template(self):
        # self.participant.vars['total_payoff'] = 1
        # self.player.final_payoff = self.participant.vars['total_payoff']
        # return {'t': self.player.final_payoff}
        return {'t': self.participant.payoff_plus_participation_fee()}
    pass


page_sequence = [
    RiskAversion,
    Demographics,
    Studies,
    Finance,
    Final
]