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


class SingleLottery(Page):
    form_model = 'player'
    form_fields = ['s_lottery', 'complex_single']

class DoubleLottery(Page):
    form_model = 'player'
    form_fields = ['d_lottery1', 'd_lottery2', 'complex_double']

class BerlinNumeracy(Page):
    form_model = 'player'
    form_fields = ['berlin_choire', 'berlin_die', 'berlin_loadedDie', 'berlin_mushrooms']

    # Before the player reaches the next page, it is evaluated which questions were answered correctly 
    def before_next_page(self):
        self.player.set_is_berlin_choire_correct()
        self.player.set_is_berlin_die_correct()
        self.player.set_is_berlin_loadedDie_correct()
        self.player.set_is_berlin_mushrooms_correct()
        self.player.set_berlin_score()

class ConsistencyLottery(Page):
    form_model = 'player'
    form_fields = ['cons_lottery1', 'cons_lottery2', 'complex_consistency']

class CombinedLottery(Page):
    form_model = 'player'
    form_fields = ['comb_lottery', 'complex_combined']

page_sequence = [SingleLottery,
                 DoubleLottery, BerlinNumeracy,
                 ConsistencyLottery, CombinedLottery]