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


class IDinput(Page):
    form_model = 'player'
    form_fields = ['ID_number']

    timeout_seconds = 600


class WaitforStart(WaitPage):
    wait_for_all_groups = True


class Startbelief(Page):
    timeout_seconds = 60


class elicitation1_A(Page):
    form_model = 'player'
    form_fields = ['DG1_belief_A']

    def is_displayed(self):
        return self.player.id_in_group == 1 or self.player.id_in_group == 3


class elicitation1_B(Page):
    form_model = 'player'
    form_fields = ['DG1_belief_B']

    def is_displayed(self):
        return self.player.id_in_group == 2 or self.player.id_in_group == 4


class elicitation2_A2(Page):
    form_model = 'player'
    form_fields = ['DG2_belief_A2']

    def is_displayed(self):
        return self.player.id_in_group == 1 or self.player.id_in_group == 3


class elicitation2_A1A2(Page):
    form_model = 'player'
    form_fields = ['DG2_belief_A1A2']

    def is_displayed(self):
        return self.player.id_in_group == 1 or self.player.id_in_group == 3


class elicitation2_A1B(Page):
    form_model = 'player'
    form_fields = ['DG2_belief_A1B']

    def is_displayed(self):
        return self.player.id_in_group == 1 or self.player.id_in_group == 3


class elicitation2_B(Page):
    form_model = 'player'
    form_fields = ['DG2_belief_B']

    def is_displayed(self):
        return self.player.id_in_group == 2 or self.player.id_in_group == 4



class AllWait(WaitPage):
    wait_for_all_groups = True


class Results(Page):
    pass


page_sequence = [WaitforStart,
                 Startbelief,
                 elicitation1_A,
                 elicitation1_B,
                 elicitation2_A2,
                 elicitation2_A1A2,
                 elicitation2_A1B,
                 elicitation2_B,
                 AllWait]