from .models import *


class Extract(Page):
    form_model = 'player'
    form_fields = ['extraction']


class MyWaitPage(WaitPage):
    wait_for_all_groups = False
    after_all_players_arrive = 'set_payoffs'


class Results(Page):
    pass


class Final_Results(Page):
    def is_displayed(player):
        return player.round_number == Constants.num_rounds


page_sequence = [Extract, MyWaitPage, Results, Final_Results]