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

class Explanation(Page):
    form_model = 'player'
    def vars_for_template(self):
        if (self.player.treatment_group == "yearly_edu" or self.player.treatment_group =="yearly"):
                roundnumber = 15
                months = 12
                frequency = "jeder"
                decision = "getroffene Entscheidung bezieht"
        else:
            roundnumber = 60
            months = 3
            frequency = "jeder vierten"
            decision = "getroffenen Entscheidungen beziehen"
        return dict(
        roundnumber = roundnumber,
        months = months,
        frequency = frequency,
        decision = decision)
class Comprehension(Page):
    form_model = 'player'
    form_fields = ['timeframe', 'portfolio_growth', 'cash_growth', 'final_payout', 'first_round']
class ComprehensionAnswers(Page):
    form_model = 'player'
    def vars_for_template(self):
        if (self.player.treatment_group == "yearly_edu" or self.player.treatment_group =="yearly"):
            is_yearly_view = True
        else:
            is_yearly_view = False
        return dict(
            is_yearly_view = is_yearly_view
        )
class Questionnaire(Page):
    form_model = 'player'
    form_fields = ['gender', 'loss_tolerance', 'advisor', 'wealth_manager', 'online_broker', 'net_monthly_income', 'available_wealth', 'age']
page_sequence = [Explanation, Comprehension, ComprehensionAnswers, Questionnaire]