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


class MyPage1(Page):
    form_model = 'player'
    form_fields = ['subject']

class MyPage2(Page):
    form_model = 'player'
    form_fields = ['subject2']

class MyPage3(Page):
    form_model = 'player'
    form_fields = ['input1_amount', 'input2_amount', 'input3_amount', 'input4_amount']

    

class ResultsWaitPage(WaitPage):

    def after_all_players_arrive(self):
        pass


class Results(Page):

    def vars_for_template(self):
        player = self.player 
        player.dis1 = ((player.input1_amount / 4000) ** 2) - 1
        player.dis2 = ((player.input2_amount / 4000) ** 1) - 1
        player.dis3 = ((player.input3_amount / 4000) ** 0.5) - 1
        player.dis4 = ((player.input4_amount / 4000) ** 0.25) - 1 
        return{
            'dis1': player.dis1,
            'dis2': player.dis2,
            'dis3': player.dis3,
            'dis4': player.dis4,
        }




page_sequence = [
    MyPage1,
    MyPage2,
    MyPage3,
    Results
]