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

class Instruct1(Page):
    def is_displayed(self):
        return self.subsession.round_number==1

class Instruct2(Page):
    def is_displayed(self):
        return self.subsession.round_number==1

class Est(Page):

    form_model = 'player'
    form_fields = ['Est']

    def vars_for_template(self):
        return {
            'image_path': 'Jars/sJar{}a1.png'.format(self.round_number),
            'round_number': self.round_number
        }

#class ResultsWaitPage(WaitPage):

    def before_next_page(self):
        self.player.payoff = max(0, self.player.session.vars['Beta'] - self.player.session.vars['Alpha'] *
         (Constants.totals[self.player.round_number-1] - self.player.Est) * (Constants.totals[self.player.round_number-1] - self.player.Est))/ Constants.num_rounds

#class Results(Page):
#    pass


page_sequence = [
    Instruct1,
    Instruct2,
    Est
#    ResultsWaitPage,
#    Results
]