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


class Einfuehrung(Page):
    pass


class Kursentwicklung_T0_T1(Page):
    form_model = 'player'
    form_fields = ['Beliefs_1', 'buy_sell_1', 'amount_buy_sell_1']

    

    def vars_for_template:
        return Constants.endowment_cash / self.player.stock_price_1
    def sell_1_max(self):
        return Constants.endowment_stock
    pass

class Tweet_1(Page):
    form_model = 'player'
    form_fields = ['Beliefs_Tweet_1', 'buy_sell_Tweet_1', 'amount_buy_sell_Tweet_1']

    def buy_Tweet_1_max(self):
        return self.player.cash_amount_1()
    def sell_Tweet_1_max(self):
        return self.player.stock_amount_1()
    pass

class Kursentwicklung_T1_T2(Page):
    form_model = 'player'
    form_fields = ['Beliefs_2', 'buy_sell_2', 'amount_buy_sell_2']

    def buy_2_max(self):
        return self.player.cash_amount_1()
    def sell_2_max(self):
        return self.player.stock_amount_1()
    pass


class Tweet_2(Page):
    form_model = 'player'
    form_fields = ['Beliefs_Tweet_2', 'buy_sell_Tweet_2', 'amount_buy_sell_Tweet_2']

    def buy_Tweet_2_max(self):
        return ((self.player.sell_2 - self.player.buy_2 + self.player.sell_Tweet_1 - self.player.buy_Tweet_1 + self.player.sell_1 - self.player.buy_1) * self.player.stock_price_2 + Constants.endowment_cash) / self.player.stock_price_2

    def sell_Tweet_2_max(self):
        return self.player.buy_2 - self.player.sell_2 + self.player.buy_Tweet_1 - self.player.sell_Tweet_1 + self.player.buy_1 - self.player.sell_1 + Constants.endowment_stock
    pass

class Kursentwicklung_T2_T3(Page):
    form_model = 'player'
    form_fields = ['Beliefs_3', 'buy_sell_3', 'amount_buy_sell_3']

    def buy_3_max(self):
        return ((self.player.sell_Tweet_2 - self.player.buy_Tweet_2 + self.player.sell_2 - self.player.buy_2 + self.player.sell_Tweet_1 - self.player.buy_Tweet_1 + self.player.sell_1 - self.player.buy_1) * self.player.stock_price_3 + Constants.endowment_cash) / self.player.stock_price_3

    def sell_3_max(self):
        return self.player.buy_Tweet_2 - self.player.sell_Tweet_2 + self.player.buy_2 - self.player.sell_2 + self.player.buy_Tweet_1 - self.player.sell_Tweet_1 + self.player.buy_1 - self.player.sell_1 + Constants.endowment_stock
    pass


class Ergebnis(Page):
    pass


page_sequence = [
    Einfuehrung,
    Kursentwicklung_T0_T1,
    Tweet_1,
    Kursentwicklung_T1_T2,
    Tweet_2,
    Kursentwicklung_T2_T3,
    Ergebnis,
]