from ._builtin import Page, WaitPage
from .models import Constants

class Welcome(Page):
    pass

class Introduction(Page):
    pass

class Familiarize(Page):
    form_model = 'player'
    form_fields = ['familiar1',
                   'familiar2',
                   ]

class Offer(Page):
    form_model = 'group'
    form_fields = ['kept']

    def is_displayed(self):
        return self.player.id_in_group == 1


class ResultsWaitPage(WaitPage):
    def after_all_players_arrive(self):
        self.group.set_payoffs()


class Results(Page):
    def before_next_page(self):
        self.player.set_payoff()

    def vars_for_template(self):
        player_in_all_rounds = self.player.in_all_rounds()
        count = [p.payoff for p in player_in_all_rounds]
        self.player.participant.vars['game1_payoff'] = count

        return {
            'offer': Constants.endowment - self.group.kept,
            'game1_payoff': count,
        }


page_sequence = [
    Welcome,
    Familiarize,
    Introduction,
    Offer,
    ResultsWaitPage,
    Results
]