from otree.api import Currency as c, currency_range

from ._builtin import Page, WaitPage
from .models import Constants


class Introduction(Page):
    pass


class RealEffortTask(Page):
    pass


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

    def before_next_page(self):
        if self.player.stock_price == "up":
            if self.player.cfs == 1:
                self.player.income = 117.07
            elif self.player.cfs == 2:
                self.player.income = 131.25
            elif self.player.cfs == 3:
                self.player.income = 135.56
            else:
                self.player.income = 135.48
        else:
            if self.player.cfs == 1:
                self.player.income = 92.68
            elif self.player.cfs == 2:
                self.player.income = 89.58
            elif self.player.cfs == 3:
                self.player.income = 91.11
            else:
                self.player.income = 70.97


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


class Reason(Page):
    form_model = 'player'
    form_fields = ['reason_self', 'reason_other']

    def before_next_page(self):
        self.player.payoff = self.player.income * 0.2


class Result(Page):
    pass


page_sequence = [Introduction, RealEffortTask, ChoiceForSelf, ChoiceForOther, Reason, Result]