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

class Period1(Page):
    form_model = 'player'
    form_fields = ['percentage']
class Period2(Page):
    form_model = 'player'
    form_fields = ['percentage2']
    def vars_for_template(self):
        securities = self.player.percentage/100 *Constants.endowment*(1 + Constants.return_securities)
        realestate = (1-self.player.percentage/100) * Constants.endowment*(1+Constants.return_realestate)
        total2=securities + realestate
        return dict(
            s2=securities,
            r2=realestate,
            total2=total2)
class Period3(Page):
    form_model = 'player'
    form_fields = ['percentage3']
    def vars_for_template(self):
        securities = self.player.percentage/100 *Constants.endowment*(1 + Constants.return_securities)
        realestate = (1-self.player.percentage/100) * Constants.endowment*(1+Constants.return_realestate)
        total2=securities + realestate
        s3 = self.player.percentage2/100 *total2*(1 + Constants.return_securities)
        r3 = (1-self.player.percentage2/100) * total2*(1+Constants.return_realestate)
        total3=s3 + r3
        return dict(
            s3=s3,
            r3=r3,
            total3=total3)
page_sequence = [Period1, Period2, Period3]