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


class Answer_A(Page):
    form_model='player'
    form_fields=['name_a','sn_a','pv_a']

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

    def after_all_players_arrive(self):
        pass

class Answer_B(Page):
    form_model='player'
    form_fields=['name_b','sn_b','pv_b']

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

class Results(Page):
    def vars_for_template(self):
        player=self.player
        player.price = 130-(player.pv_a + player.pv_b)
        player.benefit_a = player.price * player.pv_a - 10 * player.pv_a
        player.benefit_b = player.price * player.pv_b - 10 * player.pv_b

        return {
            'price':player.price,
            'benefit_a':player.benefit_a,
            'benefit_b':player.benefit_b
        }


page_sequence = [
    Answer_A,
    ResultsWaitPage,
    Answer_B,
    Results
]