from ._builtin import Page

import random


class Survey(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'nationality', 'study', 'course',
                   'question6', 'question7', 'question8', 'question9', 'question10',
                   'question11', 'question12', 'question13', 'question14']

    def before_next_page(self):

        # Temporary:
        # self.player.colors_reversed = random.randint(0, 1)
        # self.player.participant.vars['colors_reversed'] = self.player.colors_reversed

        # Temporary:
        # self.player.roles_reversed = random.randint(0, 1)
        # self.player.participant.vars['roles_reversed'] = self.player.roles_reversed

        self.player.colors_reversed = self.participant.vars.get('colors_reversed')
        self.player.roles_reversed = self.participant.vars.get('roles_reversed')


class Ranking1(Page):
    form_model = 'player'
    form_fields = ['rank1_1', 'rank1_2', 'rank1_3', 'rank1_4']

    def get_form_fields(self):

        fields = self.form_fields
        random.shuffle(fields)

        return fields


class Ranking2(Page):
    form_model = 'player'
    form_fields = ['rank2_1', 'rank2_2', 'rank2_3', 'rank2_4']

    def get_form_fields(self):

        fields = self.form_fields
        random.shuffle(fields)

        return fields


class Ranking3(Page):
    form_model = 'player'
    form_fields = ['rank3_1', 'rank3_2', 'rank3_3', 'rank3_4']

    def get_form_fields(self):

        fields = self.form_fields
        random.shuffle(fields)

        return fields


class Ranking4(Page):
    form_model = 'player'
    form_fields = ['rank4_1', 'rank4_2', 'rank4_3', 'rank4_4']

    def get_form_fields(self):

        fields = self.form_fields
        random.shuffle(fields)

        return fields


class Payment(Page):
    form_model = 'player'
    form_fields = ['first_name', 'last_name', 'address', 'city', 'country',
                   'dob', 'iban', 'email']


class End(Page):
    pass


page_sequence = [
    Survey,
    Ranking1,
    Ranking2,
    Ranking3,
    Ranking4,
    Payment,
    End
]