from ._builtin import Page
from .models import Constants

import random


class Loading(Page):
    timeout_seconds = 0.5

    def before_next_page(self):

        if self.round_number == 1:

            randomized_problems = random.sample(Constants.problems, len(Constants.problems))
            self.player.participant.vars['randomized_problems1'] = randomized_problems

            # Temporary:
            possible_orders = [13, 31]
            self.player.randomized_order = random.choice(possible_orders)
            self.player.participant.vars['randomized_order'] = self.player.randomized_order

            # 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

        problem_data = self.player.current_problem()
        self.player.problem_id = int(problem_data['id'])
        self.player.problem_type = int(problem_data['type'])
        self.player.payoff_a = int(problem_data['a'])
        self.player.payoff_b = int(problem_data['b'])
        self.player.payoff_c = int(problem_data['c'])
        self.player.payoff_d = int(problem_data['d'])
        self.player.payoff_e = int(problem_data['e'])
        self.player.payoff_f = int(problem_data['f'])
        self.player.payoff_g = int(problem_data['g'])
        self.player.payoff_h = int(problem_data['h'])

        self.player.randomized_order = self.participant.vars.get('randomized_order')

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


class Instructions1(Page):

    def is_displayed(self):

        return self.round_number == 1


class Questions1(Page):
    form_model = 'player'
    form_fields = ['q1', 'q2', 'q3']

    def is_displayed(self):

        return self.round_number == 1

    def before_next_page(self):

        if self.player.roles_reversed == 0:

            self.player.q1_error = False
            if self.player.q1 != 2:
                self.player.q1_error = True

        if self.player.roles_reversed == 1:

            self.player.q1_error = False
            if self.player.q1 != 1:
                self.player.q1_error = True

        self.player.q2_error = False
        if self.player.q2 != 1040:
            self.player.q2_error = True

        self.player.q3_error = False
        if self.player.q3 != 410:
            self.player.q3_error = True

        self.player.cq_error = False
        if self.player.q1_error or self.player.q2_error or self.player.q3_error:
            self.player.cq_error = True

        if self.player.cq_error:
            self._is_frozen = False
            self._index_in_pages -= 2
            self.participant._index_in_pages -= 2


class Instructions2(Page):

    def is_displayed(self):

        return self.round_number == 1


class Questions2(Page):
    form_model = 'player'
    form_fields = ['q4', 'q5']

    def is_displayed(self):

        return self.round_number == 1

    def q4_choices(self):

        if self.player.roles_reversed == 0:

            choices = [
                [1, 'U: 100%, D: 0%'],
                [2, 'U: 87%, D: 13%'],
                [3, 'U: 13%, D: 87%'],
                [4, 'U: 0%, D: 100%']
            ]

            return choices

        if self.player.roles_reversed == 1:

            choices = [
                [1, 'L: 100%, R: 0%'],
                [2, 'L: 87%, R: 13%'],
                [3, 'L: 13%, R: 87%'],
                [4, 'L: 0%, R: 100%']
            ]

            return choices

    def before_next_page(self):

        self.player.q4_error = False
        if self.player.q4 != 3:
            self.player.q4_error = True

        self.player.q5_error = False
        if self.player.q5 != 1:
            self.player.q5_error = True

        self.player.cq_error = False
        if self.player.q4_error or self.player.q5_error:
            self.player.cq_error = True

        if self.player.cq_error:
            self._is_frozen = False
            self._index_in_pages -= 2
            self.participant._index_in_pages -= 2


class Instructions3(Page):

    def is_displayed(self):

        return self.round_number == 1


class Questions3(Page):

    def is_displayed(self):

        return self.round_number == 1


class Questions4(Page):
    form_model = 'player'
    form_fields = ['practice_submitted_belief']

    def is_displayed(self):

        return self.round_number == 1


class Questions5(Page):

    def is_displayed(self):

        return self.round_number == 1


class Questions6(Page):
    form_model = 'player'
    form_fields = ['practice_submitted_choice']

    def is_displayed(self):

        return self.round_number == 1

    def practice_submitted_choice_choices(self):

        if self.player.roles_reversed == 0:

            choices = [
                [1, 'L'],
                [2, 'R']
            ]

            return choices

        if self.player.roles_reversed == 1:

            choices = [
                [1, 'U'],
                [2, 'D']
            ]

            return choices


class Start(Page):

    def is_displayed(self):

        return self.round_number == 1


class Problems(Page):
    form_model = 'player'
    form_fields = ['clicks_a', 'clicks_b', 'clicks_c', 'clicks_d',
                   'clicks_e', 'clicks_f', 'clicks_g', 'clicks_h',
                   'submitted_time']


class Problems1(Page):
    form_model = 'player'
    form_fields = ['submitted_choice']

    def is_displayed(self):

        return self.player.problem_type == 1

    def submitted_choice_choices(self):

        if self.player.roles_reversed == 0:

            choices = [
                [1, 'L'],
                [2, 'R']
            ]

            return choices

        if self.player.roles_reversed == 1:

            choices = [
                [1, 'U'],
                [2, 'D']
            ]

            return choices


class Problems2(Page):
    form_model = 'player'
    form_fields = ['submitted_belief']

    def is_displayed(self):

        return self.player.problem_type == 2


class End(Page):

    def is_displayed(self):

        return self.round_number == Constants.num_rounds


# CTBS2_Part1A

# def app_after_this_page(self, upcoming_apps):

# print('upcoming_apps is', upcoming_apps)

# if self.player.randomized_order == 13:
# return "CTBS2_Part3"


# CTBS2_Part1B

# def app_after_this_page(self, upcoming_apps):

# print('upcoming_apps is', upcoming_apps)

# if self.player.randomized_order == 31:
# return "CTBS2_Part4"


page_sequence = [
    Loading,
    Instructions1,
    Questions1,
    Instructions2,
    Questions2,
    Instructions3,
    Questions3,
    Questions4,
    Questions5,
    Questions6,
    Start,
    Problems,
    Problems1,
    Problems2,
    End
]