from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants


class FirstStep(Page):
    form_model = 'player'
    form_fields = [f'SP_1ST_STEP_{str(n) if n > 9 else str(0) + str(n)}' for n in range(1, 22)]

    def vars_for_template(self):
        return dict(
            payoff_list_player_A=[i for i in range(20, -1, -1)],
            payoff_list_player_B=[i for i in range(0, 21)]
        )


class SecondStep(Page):
    form_model = 'player'
    form_fields = [f'SP_2ND_STEP_{str(n) if n > 9 else str(0) + str(n)}' for n in range(1, 22)]


page_sequence = [
    FirstStep,
    SecondStep
]