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

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

class CountPref(WaitPage):
    def after_all_players_arrive(self):
        if self.round_number == 1:
            pass
        else:
            self.group.countpref()
    # pass

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

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


class ResultsWaitPage1(WaitPage):
    def after_all_players_arrive(self):
        self.group.feedback()
        # self.group.countpref()


class Results(Page):
    form_model = 'player'
    form_fields = ['good']

class ResultsWaitPage2(WaitPage):
    #  pass
    def after_all_players_arrive(self):
        self.group.feedbackgood()
        # self.group.countpref()

class ExpEnd(Page):
    def is_displayed(self):
        return self.round_number == C.NUM_ROUNDS

page_sequence = [Intro,CountPref, MyPage, ResultsWaitPage1, Results, ResultsWaitPage2, ExpEnd]