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


class Landing(Page):
    form_model = 'player'
    form_fields = ["check1", "check2", "check3"]

class Dice(Page):
    form_model = 'player'
    form_fields = ["reported_dice", "dice"]

    def before_next_page(self):
        me = self.player
        me.payoff = me.reported_dice * Constants.unit_price

class Results(Page):
    form_model = 'player'

    def vars_for_template(self):
        payoff = self.player.payoff
        return dict(
            payoff = payoff
        )

class End(Page):
    form_model = 'player'

    def is_displayed(self):
        return self.participant.vars['course']

class End_nocourse(Page):
    form_model = 'player'

    def is_displayed(self):
        return not self.participant.vars['course']

page_sequence = [Landing, Dice, Results,End,End_nocourse]