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


class MyPage(Page):
    pass


class ResultsWaitPage(WaitPage):
    pass


class Results(Page):
    pass


class end_questions(Page):
    form_model = 'player'
    form_fields = ['decision_description','job','gender','race_ethnicity','age']

class eg(Page):
    form_model = 'player'
    form_fields = ['eg_choice']


class eg_results(Page):
    def vars_for_template(self):
        me = self.player
        if me.eg_choice == 1:
                me.eg_payoff = 4
        elif me.eg_choice == 2:
            if me.eg_rand == 1:
                me.eg_payoff = 6
            else:
                me.eg_payoff = 3
        elif me.eg_choice == 3:
            if me.eg_rand == 1:
                me.eg_payoff = 8
            else:
                me.eg_payoff = 2
        elif me.eg_choice == 4:
            if me.eg_rand == 1:
                me.eg_payoff = 10
            else:
                me.eg_payoff = 1
        elif me.eg_choice == 5:
            if me.eg_rand == 1:
                me.eg_payoff = 12
            else:
                me.eg_payoff = 0
        return {'eg_payoff': me.eg_payoff}




page_sequence = [
    eg,
    eg_results,
    end_questions,
]