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


class DeterminantProbability(Page):
    form_model = models.Player
    form_fields = ['eq_prob_determinant']


class AlmostProbability(Page):
    form_model = models.Player
    form_fields = ['eq_prob_almost']


class ResultsWaitPage(WaitPage):

    def after_all_players_arrive(self):
        self.group.set_payoffs()


class Results(Page):
    def vars_for_template(self):

        return {

            'avg_det_prob': self.player.participant.vars.get('avg_prob_det'),
            'avg_alm_prob': self.player.participant.vars.get('avg_prob_alm')
        }

page_sequence = [
    DeterminantProbability,
    AlmostProbability,
    ResultsWaitPage,
    Results
]