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


class Worker_2_A_Welcome(Page):
    form_model = 'player'
    form_fields = ['prolificID']

    def before_next_page(self):
        self.player.extractmanager()


class Worker_2_B_Error(Page):
    def is_displayed(self):
        return self.player.matchedID is False


class Worker_2_C_P1_Instruction(Page):
    form_model = 'player'
    form_fields = [
        'part1ctrl1',
        'part1ctrl2',
        'part1ctrl3',
    ]

    def error_message(self, value):
        return self.player.error_message(value)


class Worker_2_D_Overview(Page):
    pass


class Worker_2_E_Comp1Decision(Page):
    form_model = 'player'
    form_fields = [
        'compchoice1',
    ]


class Worker_2_F_Feedback(Page):
    form_model = 'player'
    form_fields = [
        'posterior1',
        'posterior2',
        'posterior3',
        'posterior4',
    ]


class Worker_2_G_Comp2Decision(Page):
    form_model = 'player'
    form_fields = [
        'compchoice2',
    ]

    def is_displayed(self):
        return self.player.treatcompfirst is False


class Worker_2_H_SignalBelief(Page):
    form_model = 'player'
    form_fields = [
        'beliefsignaltype',
    ]


page_sequence = [
    Worker_2_A_Welcome,
    Worker_2_B_Error,
    Worker_2_C_P1_Instruction,
    Worker_2_D_Overview,
    Worker_2_E_Comp1Decision,
    Worker_2_F_Feedback,
    Worker_2_G_Comp2Decision,
    Worker_2_H_SignalBelief,
                 ]