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

class Instructions(Page):
    pass

class Bonus(Page):
    pass

class TheChoice(Page):
    form_model = 'player'
    form_fields = ['choices']



class Results4(Page):

        def vars_for_template(self):

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

            return dict(
                what_is_the_answer_to_choices=self.player.choices,
                x=p.lottery_outcome,
                y=p.lottery_outcome + 1.7,
                 )


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

class FinalPage(Page):
    form_model='player'
    form_fields=['probability_of_the_rare_event']

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

class AfterR(Page):
    pass

page_sequence = [
    Instructions,
    Bonus,
    TheChoice,
    Results4,
    FinalPage,
    FinalQuestion,
    AfterR
]