from otree.api import Page, WaitPage

class PrisonerDilemma(Page):
    form_model = 'player'
    form_fields = ['decision']


class ResultsWaitPage(WaitPage):
    after_all_players_arrive = 'set_payoff'

class Results(Page):
    @staticmethod
    def vars_for_template(player):
        return {
            'payoff': player.payoff
        }

page_sequence = [PrisonerDilemma, ResultsWaitPage, Results]