from .models import *

class Shuffling(WaitPage):
    wait_for_all_groups = True
    body_text = 'Groups are now shuffled. You are no more with the previous set of players.'
    def is_displayed(self):
        player = self.player
        return player.round_number == 1
class InvestmentPage(Page):
    form_model = 'player'
    form_fields = ['investment']
class WaitAllPlayers(WaitPage):
    after_all_players_arrive = 'set_payoff'
class ResultsPage(Page):
    form_model = 'player'
page_sequence = [Shuffling, InvestmentPage, WaitAllPlayers, ResultsPage]