from .models import *

class Introduction(Page):
    form_model = 'player'
    timeout_seconds = 100
class Decision(Page):
    form_model = 'player'
    form_fields = ['decision']
class ResultsWaitPage(WaitPage):
    after_all_players_arrive = 'set_payoffs'
class Results(Page):
    form_model = 'player'
    def vars_for_template(self):
        player = self.player
        me = player
        opponent = other_player(me)
        return dict(
            my_decision=me.decision,
            opponent_decision=opponent.decision,
            same_choice=me.decision == opponent.decision,
        )
page_sequence = [Introduction, Decision, ResultsWaitPage, Results]