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

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

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

class ResultsWaitPage(WaitPage):
    pass

class Results(Page):
    def vars_for_template(self):
        player = self.player 
        player.discount1 = ((player.input1_amount/4000)**(1/0.5)) -1 
        player.discount2 = ((player.input2_amount/4000)**(1/1)) -1 
        player.discount3 = ((player.input3_amount/4000)**(1/2)) -1 
        player.discount4 = ((player.input4_amount/4000)**(1/4)) -1 
        return{
            'discount1': player.discount1,
            'discount2': player.discount2,
            'discount3': player.discount3,
            'discount4': player.discount4,
        }


page_sequence = [
    MyPage1,
    MyPage2,
    MyPage3,
    ResultsWaitPage,
    Results
]