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


class Submission(Page):
    form_model = 'player'
    form_fields = ['sub1', 'sub2', 'sub3', 'sub4', 'sub5']
    def vars_for_template(self):
        print('self.player.type: ', self.player.type)
        return {'student_preference': Constants.student_preference,
                'school_priority': Constants.school_priority,
                'player_preference': Constants.student_preference[self.player.type]}


class ResultsWaitPage(WaitPage):
    def after_all_players_arrive(self):
        # boston 1-4 / da 5-8
        if self.subsession.round_number < 5:
            self.group.boston()
        else:
            self.group.da()
        self.group.set_payoff()


class Results(Page):
    pass


class Introduction(Page):
    def is_displayed(self):
        return (self.round_number == 1) or (self.round_number == 5)


page_sequence = [Introduction, Submission, ResultsWaitPage, Results]
#StartWaitingPage,