from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants
import ast
import random
import numpy as np
import math


class No1Introduction(Page):

    def is_displayed(self):
        return self.round_number == 1


class No2Instructions1(Page):

    def is_displayed(self):
        return self.round_number == 1

    def vars_for_template(self):
        return {
            'endowment': self.session.config["endowment"],
            'instrument' : "preference_discovery/SC1.jpg",
            'result': "preference_discovery/SC2.jpg",
            'show_up_fee': int(self.session.config["participation_fee"]),
        }

class No2Warning(Page):

    def is_displayed(self):
        return self.round_number == self.session.config["training_rounds"] + 1


class No3Start(Page):

    def is_displayed(self):
        return self.round_number <= self.session.config['rounds']

    def before_next_page(self):
        return {self.player.set_player_param()}

    def vars_for_template(self):
        return {
            'training': self.round_number <= self.session.config["training_rounds"],
            'training_round': self.round_number,
            'round': self.round_number - self.session.config["training_rounds"],
        }


class No4Purchase(Page):

    def is_displayed(self):
        return self.round_number <= self.session.config['rounds']

    def vars_for_template(self):
        p = self.participant.vars["displayed_prospects"]
        return {
            'p': self.participant.vars["displayed_prospects"],
            'rand_index': self.participant.vars["random_indexes"],
            'payoff_vector': self.participant.vars["payoff_vector"],
            'endowment': self.session.config["endowment"],
            'training': self.round_number <= self.session.config["training_rounds"],
            'training_round': self.round_number,
            'round': self.round_number - self.session.config["training_rounds"],
            'lot_1': p.iloc[0,2],'gain_A_1': p.iloc[0,3],'prob_A_1': p.iloc[0,4],'gain_B_1': p.iloc[0,5],'prob_B_1': p.iloc[0,6],'rel_1': p.iloc[0,7],
            'lot_2': p.iloc[1,2],'gain_A_2': p.iloc[1,3],'prob_A_2': p.iloc[1,4],'gain_B_2': p.iloc[1,5],'prob_B_2': p.iloc[1,6],'rel_2': p.iloc[1,7],
            'lot_3': p.iloc[2,2],'gain_A_3': p.iloc[2,3],'prob_A_3': p.iloc[2,4],'gain_B_3': p.iloc[2,5],'prob_B_3': p.iloc[2,6],'rel_3': p.iloc[2,7],
            'lot_4': p.iloc[3,2],'gain_A_4': p.iloc[3,3],'prob_A_4': p.iloc[3,4],'gain_B_4': p.iloc[3,5],'prob_B_4': p.iloc[3,6],'rel_4': p.iloc[3,7],
            'lot_5': p.iloc[4,2],'gain_A_5': p.iloc[4,3],'prob_A_5': p.iloc[4,4],'gain_B_5': p.iloc[4,5],'prob_B_5': p.iloc[4,6],'rel_5': p.iloc[4,7],
            'df': self.participant.vars["prospect_table"],
            'pagehold_timer': self.session.config['submit_delay'],
            'pagehold_timer_ths': self.session.config['submit_delay'] * 1000,
        }

    form_model = 'player'

    def get_form_fields(self):
        fields = ['Lotere_A', 'Lotere_B', 'Lotere_C', 'Lotere_D', 'Lotere_E']
        return fields

    def error_message(self, values):
        if values['Lotere_A'] + values['Lotere_B'] + values['Lotere_C'] + values['Lotere_D'] + values['Lotere_E'] > self.session.config["endowment"]:
            return 'Total alokasi untuk seluruh alternatif tidak boleh lebih dari ' + str(self.session.config["endowment"]) + " poin!"

    def before_next_page(self):
        return {self.player.payoff_realizer()}


class No5Result(Page):

    def is_displayed(self):
        return self.round_number <= self.session.config['rounds']

    def vars_for_template(self):
        df = self.participant.vars["displayed_prospects"][["x1","x2","Allocation","A_or_B","payoff"]]
        return {
            'training': self.round_number <= self.session.config["training_rounds"],
            'training_round': self.round_number,
            'round': self.round_number - self.session.config["training_rounds"],
            'A1': df.iloc[0, 0], 'B1': df.iloc[0, 1], 'C1': df.iloc[0, 2], 'D1': df.iloc[0, 3], 'E1': df.iloc[0, 4],
            'A2': df.iloc[1, 0], 'B2': df.iloc[1, 1], 'C2': df.iloc[1, 2], 'D2': df.iloc[1, 3], 'E2': df.iloc[1, 4],
            'A3': df.iloc[2, 0], 'B3': df.iloc[2, 1], 'C3': df.iloc[2, 2], 'D3': df.iloc[2, 3], 'E3': df.iloc[2, 4],
            'A4': df.iloc[3, 0], 'B4': df.iloc[3, 1], 'C4': df.iloc[3, 2], 'D4': df.iloc[3, 3], 'E4': df.iloc[3, 4],
            'A5': df.iloc[4, 0], 'B5': df.iloc[4, 1], 'C5': df.iloc[4, 2], 'D5': df.iloc[4, 3], 'E5': df.iloc[4, 4],
            'payoff_thisround': self.player.payoff_thisround,
        }


class No6EndQuestionnaire(Page):

    def is_displayed(self):
        return self.round_number == self.session.config['rounds']


class No6EndResult(Page):

    def is_displayed(self):
        return self.round_number == self.session.config['rounds']


page_sequence = [No1Introduction,
                 No2Instructions1,
                 No2Warning,
                 No3Start,
                 No4Purchase,
                 No5Result,
                 No6EndQuestionnaire,
                 No6EndResult]