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


class MyPage(Page):
    form_model='player'
    form_fields=['production']


class ResultsWaitPage(WaitPage):

    def after_all_players_arrive(self):
        pass


class Results(Page):
    def vars_for_template(self):
        group=self.group
        p1 = group.get_player_by_id(1)
        p2 = group.get_player_by_id(2)
        player = self.player
        group.num_production = p1.production + p2.production
        group.price = 130 - group.num_production
        player.benefit = (group.price - 10)*player.production
        return{
            'price':group.price,
            'benefit':player.benefit,
            'num_production':group.num_production
        }

class AllResultsWaitPage(WaitPage):
    wait_for_all_groups = True



page_sequence = [
    MyPage,
    ResultsWaitPage,
    Results,
    AllResultsWaitPage,
]