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


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

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

class q3(Page):
    form_model='player'
    form_fields=['input_amount_1','input_amount_2','input_amount_3','input_amount_4']



class ResultsWaitPage(WaitPage):

    def after_all_players_arrive(self):
        pass


class Results(Page):
    def vars_for_template(self):
        player = self.player
        player.dis1 = ((player.input_amount_1/4000)**2)-1
        player.dis2 = ((player.input_amount_2/4000)**1)-1
        player.dis3 = ((player.input_amount_3/4000)**0.5)-1
        player.dis4 = ((player.input_amount_4/4000)**0.25)-1
        return{
            'dis1': player.dis1,
            'dis2': player.dis2,
            'dis3': player.dis3,
            'dis4': player.dis4
        }



page_sequence = [
    q1,
    q2,
    q3,
    Results
]