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

def vars_for_all_templates(self):
    """ get variables from session configs level """
    return {
        'task': self.player.task_identifier,
        'round': self.round_number,
    }


class WelcomeRisk(Page):

    def is_displayed(self):
        return self.round_number == 1


class Risk(Page):

    form_model = models.Player
    form_fields = [
        'choice_task',
    ]

class WaitTax(Page):
    timeout_seconds = 1

    def is_displayed(self):
        return self.round_number == Constants.num_rounds

class Tax1(Page):

    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    def vars_for_template(self):
        return {
           'growth': self.player.income_growth_condition_q1
        }
    form_model = models.Player
    form_fields = [
        'income_growth_guess1',
    ]


class Tax2(Page):

    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    form_model = models.Player
    form_fields = [
        'income_growth_guess2',
    ]

    def vars_for_template(self):
        return {
           'growth': self.player.income_growth_condition_q2
        }


class Control(Page):

    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    form_model = models.Player
    form_fields = [
        'control_probability',
    ]

class Wait(Page):
    timeout_seconds = 1


page_sequence = [
    WelcomeRisk,
    Risk,
    # Wine,
    Tax1,
    WaitTax,
    Wait,
    Tax2,
    Control,
]