from ._builtin import Page, WaitPage
from .models import Constants
from operator import attrgetter
import datetime


class Welcome(Page):
    pass


class NameA(Page):
    form_model = 'player'
    form_fields = ['nameA']

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

    def before_next_page(self):
        self.participant.vars['names'] = self.player.nameA


class Name2A(Page):
    form_model = 'player'
    form_fields = ['nameA']

    def is_displayed(self):
        return self.player.id_in_group == 1 and self.player.truth == 0

    def before_next_page(self):
        self.participant.vars['names'] = self.player.nameA


class NameB(Page):
    form_model = 'player'
    form_fields = ['nameB']

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

    def before_next_page(self):
        self.participant.vars['names'] = self.player.nameB


class Name2B(Page):
    form_model = 'player'
    form_fields = ['nameB']

    def is_displayed(self):
        return self.player.id_in_group == 2 and self.player.truth == 0

    def before_next_page(self):
        self.participant.vars['names'] = self.player.nameB

class Truth(Page):
    form_model = 'player'
    form_fields = ['truth']


class Truth2(Page):
    form_model = 'player'
    form_fields = ['truth']

    def is_displayed(self):
        return self.player.truth == 0


class WelcomeWaitPage(WaitPage):
    title_text = "Please wait"
    body_text = "Waiting for the other participants."


page_sequence = [
    Welcome,
    NameA,
    NameB,
    Truth,
    Name2A,
    Name2B,
    Truth2
]