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


class Introduction(Page):
    def is_displayed(self):
        return (self.round_number == 1)
    pass

class Example(Page):
    """
    def vars_for_template(self):
        return {'bagType': self.subsession.bag_type,'drawOne': self.player.first_draw, 'drawTwo': self.player.second_draw}
        """

class Test(Page):
    form_model = 'player'
    form_fields = ['probReservation_amount','touched_slider']
    def touched_slider_error_message(self, value):
        if not value:
            return 'Please adjust slider to chosen amount. If you desire 50, please move and return slider to 50'

    def before_next_page(self):
        self.player.draw_from_box = self.subsession.nickels_in_box >= self.player.probReservation_amount

class Results(Page):
    def vars_for_template(self):
        if self.player.draw_from_box:
            if self.subsession.nickel_drawn == "was":
                won = Constants.additional_payment
            else:
                won = 0
        else:
            if self.subsession.bag_type == "RED":
                won = Constants.additional_payment
            else:
                won = 0
        return {'won': won}


page_sequence = [
    Introduction,
    Example,
    Test,
    Results
]