from .models import *

class Trustor(Page):
    form_model = 'player'
    form_fields = ['allocate']
    def is_displayed(self):
        group = self.group
        player = self.player
        return player.id_in_group == 1
class WaitPage_1(WaitPage):
    after_all_players_arrive = 'allocate_to_trustee'
class Trustee(Page):
    form_model = 'player'
    form_fields = ['return_back']
    def is_displayed(self):
        group = self.group
        player = self.player
        return player.id_in_group == 2
class WaitPage_2(WaitPage):
    after_all_players_arrive = 'return_back_to_trustor'
class Result(Page):
    form_model = 'player'
page_sequence = [Trustor, WaitPage_1, Trustee, WaitPage_2, Result]