from ._builtin import Page
from .models import Constants


class InstructionBasePage(Page):
    timeout_seconds = Constants.timeout

    def is_displayed(self):
        return self.round_number == 1

class Gameplay(InstructionBasePage):
    def vars_for_template(self):
        return {
            'game_play_timeout_seconds': self.session.config['game_play_timeout_seconds'],
            'wrong_answer_freeze_time': self.session.config['wrong_answer_freeze_time'],
            'enforce_wait_time': self.session.config['enforce_wait_time'] * 1000
        }


class Study(InstructionBasePage):
    def vars_for_template(self):
        return {
            'periods': self.session.config['periods'],
            'rounds': self.session.config['rounds'],
            'timeout_seconds': self.session.config['timeout_seconds'],
            'population_size': self.session.config['population_size'],
            'enforce_wait_time': self.session.config['enforce_wait_time'] * 1000
        }


page_sequence = [
    Gameplay, Study
]