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


class MyPage(Page):
    pass


class Media_intro(Page):
    pass


class Sample1Neutral(Page):
    def is_displayed(self):
        if self.player.id_in_group == 1:
            return True


class Sample1Disc(Page):
    def is_displayed(self):
        if self.player.id_in_group == 2:
            return True


class Sample1Counter(Page):
    def is_displayed(self):
        if self.player.id_in_group == 3:
            return True


class Questions1(Page):
    form_model = 'player'
    form_fields = [
        's1q1',
        's1q2',
    ]


class GuessTreatment1(Page):
    form_model = 'player'
    form_fields = ['treatment1']


class Sample2Neutral(Page):
    def is_displayed(self):
        if self.player.id_in_group == 3:
            return True


class Sample2Disc(Page):
    def is_displayed(self):
        if self.player.id_in_group == 1:
            return True


class Sample2Counter(Page):
    def is_displayed(self):
        if self.player.id_in_group == 2:
            return True


class Questions2(Page):
    form_model = 'player'
    form_fields = [
        's2q1',
        's2q2',
    ]


class GuessTreatment2(Page):
    form_model = 'player'
    form_fields = ['treatment2']


class Sample3Neutral(Page):
    def is_displayed(self):
        if self.player.id_in_group == 3:
            return True


class Sample3Disc(Page):
    def is_displayed(self):
        if self.player.id_in_group == 1:
            return True


class Sample3Counter(Page):
    def is_displayed(self):
        if self.player.id_in_group == 2:
            return True


class Questions3(Page):
    form_model = 'player'
    form_fields = [
        's3q1',
        's3q2',
    ]


class GuessTreatment3(Page):
    form_model = 'player'
    form_fields = ['treatment3']


class Sample4Neutral(Page):
    def is_displayed(self):
        if self.player.id_in_group == 2:
            return True


class Sample4Disc(Page):
    def is_displayed(self):
        if self.player.id_in_group == 3:
            return True


class Sample4Counter(Page):
    def is_displayed(self):
        if self.player.id_in_group == 1:
            return True


class Questions4(Page):
    form_model = 'player'
    form_fields = [
        's4q1',
        's4q2',
    ]


class GuessTreatment4(Page):
    form_model = 'player'
    form_fields = ['treatment4']


class Sample5Neutral(Page):
    def is_displayed(self):
        if self.player.id_in_group == 1:
            return True


class Sample5Disc(Page):
    def is_displayed(self):
        if self.player.id_in_group == 2:
            return True


class Sample5Counter(Page):
    def is_displayed(self):
        if self.player.id_in_group == 3:
            return True


class Questions5(Page):
    form_model = 'player'
    form_fields = [
        's5q1',
        's5q2',
    ]


class GuessTreatment5(Page):
    form_model = 'player'
    form_fields = ['treatment5']


class PictureTime(Page):
    form_model = 'player'
    form_fields = ['pictures']


class MyWaitPage(WaitPage):
    body_text = "Please wait while everyone finishes the media stage."

    def after_all_players_arrive(self):
        for each in self.group.get_players():
            print("calculating payoffs")
            each.set_payoffs()


class MediaResult(Page):
    pass


class FinalComment(Page):
    form_model = 'player'
    form_fields = ['final_comment']


page_sequence = [
    MyPage,
    Media_intro,
    Sample1Neutral,
    Sample1Disc,
    Sample1Counter,
    Questions1,
    GuessTreatment1,
    Sample2Neutral,
    Sample2Disc,
    Sample2Counter,
    Questions2,
    GuessTreatment2,
    Sample3Neutral,
    Sample3Disc,
    Sample3Counter,
    Questions3,
    GuessTreatment3,
    Sample4Neutral,
    Sample4Disc,
    Sample4Counter,
    Questions4,
    GuessTreatment4,
    Sample5Neutral,
    Sample5Disc,
    Sample5Counter,
    Questions5,
    GuessTreatment5,
    PictureTime,
    MyWaitPage,
    MediaResult,
]