from ._builtin import Page, WaitPage
from .models import Constants


class WaitTest(WaitPage):

    def after_all_players_arrive(self):
        pass


class _quiz1(Page):
    form_model = 'player'
    form_fields = ['q1']

    def error_message(self, values):
        print('values is', values)
        if values["q1"] != self.session.vars['q1']:
            self.player.errors += 1
            return '问题1回答错误'

    def vars_for_template(self):
        return dict(
            treatment=self.session.vars['treatment'],
        )


class _quiz2(Page):
    form_model = 'player'
    form_fields = ['q2']

    def error_message(self, values):
        print('values is', values)
        if values["q2"] != self.session.vars['q2']:
            self.player.errors += 1
            return '问题2回答错误'

    def vars_for_template(self):
        return {'treatment': self.session.vars['treatment'],
                }


class _quiz3(Page):
    form_model = 'player'
    form_fields = ['q3']

    def error_message(self, values):
        print('values is', values)
        if values["q3"] != self.session.vars['q3']:
            self.player.errors += 1
            return '问题3回答错误'

    def vars_for_template(self):
        return dict(
            treatment=self.session.vars['treatment'],
        )


class _quiz4(Page):
    form_model = 'player'
    form_fields = ['q4']

    def error_message(self, values):
        print('values is', values)
        if values["q4"] != self.session.vars['q4']:
            self.player.errors += 1
            return '问题4回答错误'

    def vars_for_template(self):
        return dict(
            treatment=self.session.vars['treatment'],
        )


class _quiz5(Page):
    form_model = 'player'
    form_fields = ['q5']

    def error_message(self, values):
        print('values is', values)
        if values["q5"] != self.session.vars['q5']:
            self.player.errors += 1
            return '问题5回答错误'

    def vars_for_template(self):
        return {'treatment': self.session.vars['treatment'],
                }


class _quiz6(Page):
    form_model = 'player'
    form_fields = ['q6']

    def error_message(self, values):
        print('values is', values)
        if values["q6"] != self.session.vars['q6']:
            self.player.errors += 1
            return '问题6回答错误'

    def vars_for_template(self):
        return dict(
            treatment=self.session.vars['treatment'],
        )


class _quiz7(Page):
    form_model = 'player'
    form_fields = ['q7']

    def is_displayed(self):
        return self.subsession.treatment == 2 or self.subsession.treatment == 3 or self.subsession.treatment == 4

    def error_message(self, values):
        print('values is', values)
        if values["q7"] != self.session.vars['q7']:
            self.player.errors += 1
            return '问题7回答错误'

    def vars_for_template(self):
        return {'treatment': self.session.vars['treatment'],
                }


class _quiz72(Page):
    form_model = 'player'
    form_fields = ['q72']

    def is_displayed(self):
        return self.subsession.treatment == 5 or self.subsession.treatment == 6

    def error_message(self, values):
        print('values is', values)
        if values["q72"] != self.session.vars['q72']:
            self.player.errors += 1
            return '问题7回答错误'

    def vars_for_template(self):
        return {'treatment': self.session.vars['treatment'],
                }


class _quiz8(Page):
    form_model = 'player'
    form_fields = ['q8']

    def is_displayed(self):
        return self.subsession.treatment == 3 or self.subsession.treatment == 6

    def error_message(self, values):
        print('values is', values)
        if values["q8"] != self.session.vars['q8']:
            self.player.errors += 1
            return '问题8回答错误'

    def vars_for_template(self):
        return {'treatment': self.session.vars['treatment'],
                }


class _quiz82(Page):
    form_model = 'player'
    form_fields = ['q82']

    def is_displayed(self):
        return self.subsession.treatment == 4

    def error_message(self, values):
        print('values is', values)
        if values["q82"] != self.session.vars['q82']:
            self.player.errors += 1
            return '问题8回答错误'

    def vars_for_template(self):
        return {'treatment': self.session.vars['treatment'],
                }


class _Intro(Page):
    pass


page_sequence = [
    WaitTest,
    _quiz1,
    _quiz2,
    _quiz3,
    _quiz4,
    _quiz5,
    _quiz6,
    _quiz7,
    _quiz72,
    _quiz8,
    _quiz82,
    _Intro,
]