from ._builtin import Page

import random


class Part3A(Page):
    form_model = 'player'
    form_fields = ['statement1', 'statement2', 'statement3', 'statement4', 'statement5',
                   'statement6', 'statement7', 'statement8', 'statement9', 'statement10',
                   'statement11', 'statement12', 'statement13', 'statement14', 'statement15',
                   'statement16', 'statement17', 'statement18', 'statement19', 'statement20',
                   'statement21', 'statement22']

    def before_next_page(self):

        self.player.treatment = random.randint(1, 2)
        # self.player.treatment = self.player.participant.vars.get('treatment')


class Part3B(Page):
    pass


class Part3B1(Page):
    form_model = 'player'
    form_fields = ['rank1', 'rank2', 'rank3', 'rank4']

    def get_form_fields(self):

        fields = self.form_fields
        random.shuffle(fields)
        return fields


class Part3B2(Page):
    form_model = 'player'
    form_fields = ['choice']


class Part3B3(Page):
    form_model = 'player'
    form_fields = ['belief']


class Part3C(Page):
    form_model = 'player'
    form_fields = ['question1', 'question2', 'question3', 'question4', 'question5',
                   'question6', 'question7', 'question8', 'question9', 'question10',
                   'question11', 'question12', 'question13']


class End(Page):

    def before_next_page(self):

        self.player.participant_vars_dump = str(self.player.participant.vars)


page_sequence = [
    Part3A,
    Part3B,
    Part3B1,
    Part3B2,
    Part3B3,
    Part3C,
    End
]