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


class Questionnaire(Page):
    form_model = 'player'
    form_fields = ['salary_meaning_question', 'competitiveness_question', 'q1', 'q2', 'q3', 'q4', 'q5', 'q6', 'q7']

    def vars_for_template(self):
        return {'company': self.player.rated_company,
                'company_description': self.player.company_description
        }


class ResultsWaitPage(WaitPage):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds

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


page_sequence = [
    Questionnaire,
    ResultsWaitPage,
]