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

class ChoicePage(Page):
    form_model = 'player'
    form_fields = ['lottery_choice']


class Results(Page):
    def vars_for_template(self):

        p = self.player
        lottery2 = [0, 2, 2, 2, 2]
        if p.lottery_choice == 2:
            p.lottery_outcome = 1.5
        else:
            p.lottery_outcome = random.choice(lottery2)

        return dict(
            z=p.lottery_outcome,
        )

    def before_next_page(self):
        self.player.payoff = self.player.lottery_outcome

class AfterR(Page):

    def vars_for_template(self):

        return dict(
            w=self.participant.payoff + 0.5,
        )
class FinalPage(Page):
    form_model='player'
    form_fields=['probability_of_the_rare_event']

class FinalQuestion(Page):
    form_model='player'
    form_fields=['X']

class TheEnd(Page):
    pass
page_sequence = [ChoicePage, Results, AfterR, FinalPage, FinalQuestion, TheEnd]