from otree.api import Currency as c, currency_range

from ._builtin import Page, WaitPage
from .models import Constants


class Instructions(Page):
    form_model = 'player'

    def vars_for_template(self):
        return {'endowment': self.player.initial}


class Demographics(Page):
    form_model = 'player'
    form_fields = ['age',
                   'gender']

class PracticeRound(Page):
    form_model = 'player'
    form_fields = ['Q1']

class PracticeRoundAnswer(Page):
    form_model = 'player'

class Task1(Page):
    form_model = 'player'
    form_fields = ['Q1']
    def before_next_page(self):
        if self.player.Q1 == 16:
            self.player.Q1_correct = True

class Task2(Page):
    form_model = 'player'
    form_fields = ['Q2']
    def before_next_page(self):
        if self.player.Q2 == 39:
            self.player.Q2_correct = True

class Task3(Page):
    form_model = 'player'
    form_fields = ['Q3']
    def before_next_page(self):
        if self.player.Q3 == 69:
            self.player.Q3_correct = True

class Task4(Page):
    form_model = 'player'
    form_fields = ['Q4']
    def before_next_page(self):
        if self.player.Q4 == 70:
            self.player.Q4_correct = True

class Task5(Page):
    form_model = 'player'
    form_fields = ['Q5']
    def before_next_page(self):
        if self.player.Q5 == 119:
            self.player.Q5_correct = True

class Task6(Page):
    form_model = 'player'
    form_fields = ['Q6']
    def before_next_page(self):
        if self.player.Q6 == 120:
            self.player.Q6_correct = True

class Task7(Page):
    form_model = 'player'
    form_fields = ['Q7']
    def before_next_page(self):
        if self.player.Q7 == 155:
            self.player.Q7_correct = True

class Task8(Page):
    form_model = 'player'
    form_fields = ['Q8']
    def before_next_page(self):
        if self.player.Q8 == 232:
            self.player.Q8_correct = True

class Task9(Page):
    form_model = 'player'
    form_fields = ['Q9']
    def before_next_page(self):
        if self.player.Q9 == 313:
            self.player.Q9_correct = True

class Task10(Page):
    form_model = 'player'
    form_fields = ['Q10']
    def before_next_page(self):
        if self.player.Q10 == 449:
            self.player.Q10_correct = True

# class Wait(WaitPage):
#     def after_all_players_arrive(self):


class Results(Page):
    form_model = 'player'


page_sequence = [
    Instructions,
    Demographics,
    PracticeRound,
    PracticeRoundAnswer,
    Task1,
    Task2,
    Task3,
    Task4,
    Task5,
    Task6,
    Task7,
    Task8,
    Task9,
    Task10,
    Results
]