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 = ['production2']


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_production2 = (p1.production2 + p2.production2)
        group.price2 = (130 - group.num_production2)
        player.profit2 = (group.price2 - 10) * player.production2
        return{
            'price2': group.price2,
            'profit2': player.profit2,
            'num_production2': group.num_production2,
        }

page_sequence = [
    MyPage,
    ResultsWaitPage,
    Results,
]