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

class Comprehension(Page):
    form_model = 'player'
    form_fields = ['test']

    def error_message(self, values):
        print('values is', values)
        if values['test'] != 10:
            return 'Please correct your answer'



    def before_next_page(self):
        self.player.trole=self.group.id_in_subsession % 2

class Decision2(Page):
    form_model = 'player'
    form_fields = ['Beauty_contest', 'Payoff_choice']

    def is_displayed(self):
        return self.player.trole==0

class ResultsWaitPage(WaitPage):
    body_text = "Waiting for the other participant to decide."

    after_all_players_arrive = 'set_payoffs'

    def is_displayed(self):
        return self.player.trole==0

class Results(Page):
    def vars_for_template(self):
        return dict(Beauty_contest=round(self.player.Beauty_contest, 2),
                    mean_payoff=round(self.group.mean_payoff, 2),
                    d=round(self.player.d, 2),
                    payoff=round(self.player.payoff, 2)
                    )

    def is_displayed(self):
        return self.player.trole==0


page_sequence = [Comprehension,
                 Decision2,
                 ResultsWaitPage,
                 Results]