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


class Intro(Page):
    form_model = 'player'
    form_fields = ['consent_btn']

    def is_displayed(self):
        return self.round_number == 1


class Quiz(Page):
    form_model = 'player'
    form_fields = [
        'item1A',
        'item2A'
    ]

    def get_form_fields(self):
        fields = self.form_fields
        return fields

    def is_displayed(self):
        return self.round_number == 1


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


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


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


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


class StartWaitPage(WaitPage):
    def is_displayed(self):
        return self.round_number == 1


class MyPage(Page):
    form_model = 'player'
    form_fields = ['decision']


class ResultsWaitPage(WaitPage):
    after_all_players_arrive = 'set_payoffs'


class Results(Page):
    pass

class totalpayoff(WaitPage):
    def after_all_players_arrive(self):
        if self.round_number == 2:
            self.group.total_payoff()
        else: pass


class Finish(Page):
    def is_displayed(self):
        return self.round_number == 2




#page_sequence = [Intro, Explain, Quiz, Quiz_Answer, Explain_control, Start_page, StartWaitPage, MyPage, ResultsWaitPage, Results, totalpayoff, Finish]
page_sequence = [Start_page, StartWaitPage, MyPage, ResultsWaitPage, Results, totalpayoff, Finish]