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

class Consent(Page):
    form_model = 'player'
    def is_displayed(self):
        return self.subsession.round_number == 1
class Introduction(Page):
    form_model = 'player'
    def is_displayed(self):
        return self.subsession.round_number == 1
class Likelihood(Page):
    form_model = 'player'
    form_fields = ['times', 'most_likely', 'least_likely']
    def is_displayed(self):
        return self.subsession.round_number == 1
class ChooseSite(Page):
    form_model = 'player'
    form_fields = ['next_site']
    def before_next_page(self):
        self.player.get_payoff()
class ShowPayoff(Page):
    form_model = 'player'
class End(Page):
    form_model = 'player'
    def is_displayed(self):
        return self.subsession.round_number == Constants.num_rounds
    def before_next_page(self):
        self.participant.vars["ranger_site"] = 0
        self.participant.vars["ranks"] = [100] * 4
        self.participant.payoff = 0
page_sequence = [Consent, Introduction, Likelihood, ChooseSite, ShowPayoff, End]