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

class InstructionsEsp(Page):
    def is_displayed(self):
        return self.round_number==1

class Introduction(Page):
    def is_displayed(self):
        return self.round_number==1

class InstructionsGen(Page):
    def is_displayed(self):
        return self.round_number==1


class Offer(Page):
    form_model = 'group'
    form_fields = ['sent_amount']

    def is_displayed(self):
        return self.player.id_in_group == 1
    def vars_for_template(self):
        mostrar_ronda = True

        if self.subsession.round_number == 1:
            mostrar_ronda = False

        return dict(mostrar=mostrar_ronda)


class ExpectationsB(Page):
    form_model = 'group'
    form_fields = ['expectation_sent']
    def is_displayed(self):
        return self.player.id_in_group == 2
    def vars_for_template(self):
        mostrar_ronda = True

        if self.subsession.round_number == 1:
            mostrar_ronda = False

        return dict(mostrar=mostrar_ronda)



class RoundResults(Page):
    def vars_for_template(self):
        mostrar_ronda = True
        if self.subsession.round_number == 1:
            mostrar_ronda = False
        return dict(round_paymentA=Constants.endowmentA-self.group.sent_amount,mostrar=mostrar_ronda)
    def before_next_page(self):
        group= self.group
        playerA= group.get_player_by_id(1)
        playerB= group.get_player_by_id(2)
        playerC= group.get_player_by_id(3)


        playerA.round_payoff=(Constants.endowmentA-group.sent_amount)*2
        playerB.round_payoff=(group.sent_amount)*2
        playerC.round_payoff=(Constants.endowmentC)*2

        playerB.expected_sent_ammount_B=group.expectation_sent

        playerB.global_payoff=group.sent_amount*2
        Groups = self.subsession.get_groups()

        ##Creencias para la cantidad enviada por A.
        values_sent_ammount = []
        for g in Groups:
            values_sent_ammount.append(g.sent_amount)
        # Vector de modas
        repetitions_1 = 0
        for i in values_sent_ammount:
            aparitions = values_sent_ammount.count(i)
            if aparitions > repetitions_1:
                repetitions_1 = aparitions
        modes_sent_ammount = []
        for i in values_sent_ammount:
            aparitions = values_sent_ammount.count(i)
            if aparitions == repetitions_1 and i not in modes_sent_ammount:
                modes_sent_ammount.append(i)
        if playerB.expected_sent_ammount_B in modes_sent_ammount:
                playerB.bel_sent=True
                playerB.global_payoff += c(3)*2




class DespuesRondaUno(Page):
    def is_displayed(self):
        return self.round_number==1

class ResultsWaitPage(WaitPage):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    def after_all_players_arrive(self):
        num_round = self.session.vars['paying_round']
        group = self.group
        player_A = group.get_player_by_id(1)
        player_B = group.get_player_by_id(2)
        player_C = group.get_player_by_id(3)

        player_A.payoff=player_A.in_round(num_round).round_payoff
        player_B.payoff=player_B.in_round(num_round).global_payoff
        player_C.payoff=player_C.in_round(num_round).round_payoff


        Groups=self.subsession.get_groups()
        ##Creencias para B en la última ronda (normas sociales)
        matching = random.randint(0, len(Groups) - 2)

        ##pregunta 1
        norms_B_s1 = []
        for g in Groups:
            norms_B_s1.append(g.expectation_social_norm_s1_B)
        norms_B_s1.remove(group.expectation_social_norm_s1_B)
        if norms_B_s1[matching] == group.expectation_social_norm_s1_B:
            player_B.payoff += c(1)*2


        ##Creencias para C en la última ronda (normas sociales)
        ##pregunta 1
        norms_C_s1 = []
        for g in Groups:
            norms_C_s1.append(g.expectation_social_norm_s1_C)
        norms_C_s1.remove(group.expectation_social_norm_s1_C)
        if norms_C_s1[matching] == group.expectation_social_norm_s1_C:
            player_C.payoff += c(1)*2



class Survey(Page):
    form_model = 'player'
    form_fields = ["survey_4","survey_5","survey_6"]
    def is_displayed(self):
        return self.round_number==1

class AfterSurvey(Page):
    def vars_for_template(self):
        total_correct = 0
        correct_4 = "No"
        if self.player.survey_4 == 60:
            correct_4 = "Si"
            total_correct += 1
        correct_5 = "No"
        if self.player.survey_5 == 40:
            correct_5 = "Si"
            total_correct += 1
        correct_6 = "No"
        if self.player.survey_6 == 50:
            correct_6 = "Si"
            total_correct += 1
        show = False
        if total_correct < 3:
            show = True
        return dict(
                correct_4=correct_4, correct_5=correct_5,
                correct_6=correct_6,  total_correct=total_correct,
                show=show
        )

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


class Results(Page):

    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    def vars_for_template(self):
        group= self.group
        playerA= group.get_player_by_id(1)
        playerB= group.get_player_by_id(2)
        playerC= group.get_player_by_id(3)

        num_round = self.session.vars['paying_round']
        payoff_B= playerB.in_round(num_round).round_payoff/2

        # método para coger el vector de los valores de mensaje para la ronda escogida

        Groups = self.subsession.get_groups()
        ##Creencias para la cantidad enviada por A.
        payment_message_deduction = 0
        if playerB.in_round(num_round).bel_sent==True:
            payment_message_deduction += 3
        else:
            payment_message_deduction+=0

        ##Creencias para B en la última ronda (normas sociales)
        matching = random.randint(0, len(Groups) - 2)
        social_norms_B = c(0)

        ##pregunta 1
        norms_B_s1 = []
        correct_1_B = "no"
        for g in Groups:
            norms_B_s1.append(g.expectation_social_norm_s1_B)
        norms_B_s1.remove(self.group.expectation_social_norm_s1_B)
        if norms_B_s1[matching] == self.group.expectation_social_norm_s1_B:
            social_norms_B = +c(1)
            correct_1_B = "Sí"



        ##Creencias para C en la última ronda (normas sociales)
        social_norms_C = c(0)
        ##pregunta 1
        norms_C_s1 = []
        correct_1_C = "No"
        for g in Groups:
            norms_C_s1.append(g.expectation_social_norm_s1_C)
        norms_C_s1.remove(self.group.expectation_social_norm_s1_C)
        if norms_C_s1[matching] == self.group.expectation_social_norm_s1_C:
            social_norms_C += c(1)
            correct_1_C = "Sí"

        ##Organización en Labels para la interfaz del jugador.
        # Normas sociales en la pregunta 1 para B
        if self.group.expectation_social_norm_s1_B == 0:
            belief_sn_B1 = "Muy socialmente inapropiada"
        elif self.group.expectation_social_norm_s1_B == 1:
            belief_sn_B1 = "Algo socialmente inapropiada"
        elif self.group.expectation_social_norm_s1_B == 2:
            belief_sn_B1 = "Algo socialmente apropiada"
        else:
            belief_sn_B1 = "Muy socialmente apropiada"



        # Normas sociales en la pregunta 1 para C
        if self.group.expectation_social_norm_s1_C == 0:
            belief_sn_C1 = "Muy socialmente inapropiada"
        elif self.group.expectation_social_norm_s1_C == 1:
            belief_sn_C1 = "Algo socialmente inapropiada"
        elif self.group.expectation_social_norm_s1_C == 2:
            belief_sn_C1 = "Algo socialmente apropiada"
        else:
            belief_sn_C1 = "Muy socialmente apropiada"

        # Invocamos al "grupo" en la ronda aleatoria.
        player = playerB.in_round(num_round)
        # Sobre las expectativas de B de los puntos envíados por A.
        if player.expected_sent_ammount_B == c(0):
            belief_sent = "0 puntos"
        elif player.expected_sent_ammount_B == c(10):
            belief_sent = "10 puntos"
        elif player.expected_sent_ammount_B == c(20):
            belief_sent = "20 puntos"
        elif player.expected_sent_ammount_B == c(30):
            belief_sent = "30 puntos"
        elif player.expected_sent_ammount_B == c(40):
            belief_sent = "40 puntos"
        elif player.expected_sent_ammount_B:
            belief_sent = "50 puntos"



# Modas de los puntos enviados por A
        modes_sent_message = "No"
        if self.player.in_round(num_round).bel_sent==True:
            modes_sent_message= "Sí"

        player_A_payoff=int(playerA.payoff)/2
        player_B_payoff=int(playerB.payoff)/2
        player_C_payoff=int(playerC.payoff)/2

        return dict(
            paying_round=num_round,
            payoff_B=payoff_B,
            payment_message_deduction=payment_message_deduction,
            player_A_payoff=player_A_payoff,
            player_B_payoff=player_B_payoff,
            player_C_payoff=player_C_payoff,
            social_norms_C=social_norms_C,
            social_norms_B=social_norms_B,
            correct_1_C=correct_1_C,
            correct_1_B=correct_1_B,
            modes_sent=modes_sent_message,
            belief_sent=belief_sent,
            belief_sn_B1=belief_sn_B1,
            belief_sn_C1=belief_sn_C1,
        )


class WaitForA(WaitPage):
    wait_for_all_groups = True
    title_text="Por favor espere a que el resto de participantes tomen sus decisiones."
    body_text ="De vez en cuando (cada 30 segundos aproximadamente), por favor oprima el botón de teclado F5 para refrescar la conexión de su computador con el servidor del experimento, y así poder avanzar más rápidamente."


class WaitForAll(WaitPage):
    wait_for_all_groups = True
    title_text="Por favor espere a que el resto de participantes tomen sus decisiones."
    body_text ="De vez en cuando (cada 30 segundos aproximadamente), por favor oprima el botón de teclado F5 para refrescar la conexión de su computador con el servidor del experimento, y así poder avanzar más rápidamente."

class Survey2(Page):
    def is_displayed(self):
        total_correct = 0

        if self.player.survey_4 == 60:
            total_correct += 1
        if self.player.survey_5 == 40:
            total_correct += 1
        if self.player.survey_6 == 50:
            total_correct += 1
        show = False
        if total_correct < 3:
            show = True

        return show and self.round_number==1

    form_model = "player"
    form_fields = ["survey_8"]
    def error_message(self, values):
        if  values["survey_8"]!=65:
            return "Tiene un error en la respuesta."

class InstructionsLastRoundB(Page):
    def is_displayed(self):
        return self.round_number==Constants.num_rounds and self.player.id_in_group==2


class InstructionsLastRoundC(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds and self.player.id_in_group == 3

class PcProcedure(Page):
    def is_displayed(self):
        return self.round_number==1
class SocialNormsB(Page):
    form_model='group'
    form_fields=["expectation_social_norm_s1_B"]
    def is_displayed(self):
        return self.round_number==Constants.num_rounds and self.player.id_in_group==2

class SocialNormsC(Page):
    form_model = 'group'
    form_fields = ["expectation_social_norm_s1_C"]
    def is_displayed(self):
        return self.round_number==Constants.num_rounds and self.player.id_in_group==3

class Payment(Page):
    def vars_for_template(self):
        group = self.group
        player_A = group.get_player_by_id(1)
        player_B = group.get_player_by_id(2)
        player_C = group.get_player_by_id(3)
        player_A_payoff=player_A.payoff/2
        player_B_payoff=player_B.payoff/2
        player_C_payoff=player_C.payoff/2
        payment=self.participant.payoff.to_real_world_currency(self.session)
        total=self.participant.payoff_plus_participation_fee()

        return dict(player_A_payoff=player_A_payoff,
            player_B_payoff=player_B_payoff,
            player_C_payoff=player_C_payoff,
                    payment=payment,
                    total=total)
    def is_displayed(self):
        return self.round_number == Constants.num_rounds

class Questionnaire_A(Page):
    form_model = "player"
    form_fields = ["q1","q2","q3","q4","q5","q6"]
    def is_displayed(self):
        return self.round_number==Constants.num_rounds

class Questionnaire_B(Page):
    form_model = "player"
    form_fields = ["q7","q8","q9","q10","q11","q12","q13","q14","q15"]
    def is_displayed(self):
        return self.round_number==Constants.num_rounds

class FinalMessage(Page):
    def is_displayed(self):
        return self.round_number==Constants.num_rounds


class AllGroupsWaitPage(WaitPage):
    wait_for_all_groups = True
    def is_displayed(self):
        return self.round_number== Constants.num_rounds

class PagosInicial(Page):
    def is_displayed(self):
        return self.round_number==Constants.num_rounds

class Pagos_1(Page):
    form_model = 'player'
    form_fields = ["idcard1", "expday", "expmonth", "expyear", "namep", "address", "depto", "cityp", "country1", "email", "telephone", "dayp", "monthp", "yearp", "signaturep", "typepayment","idcard2"]

    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    def vars_for_template(self):
        group = self.group
        player_A = group.get_player_by_id(1)
        player_B = group.get_player_by_id(2)
        player_C = group.get_player_by_id(3)
        player_A_payoff = int(player_A.payoff)/2
        player_B_payoff = int(player_B.payoff) / 2
        player_C_payoff = int(player_C.payoff) / 2
        payment = self.participant.payoff.to_real_world_currency(self.session)
        total = self.participant.payoff_plus_participation_fee()
        return dict(player_A_payoff=player_A_payoff,
                    player_B_payoff=player_B_payoff,
                    player_C_payoff=player_C_payoff,
                    payment=payment,
                    total=total)


class Pagos_2(Page):
    form_model= 'player'
    def is_displayed(self):
        return self.round_number==Constants.num_rounds

    def get_form_fields(self):
        if self.player.typepayment:
            return ['bank','id_bank','name_2','account_type','idcard3']
        else:
            return ['platform','telephone_2','name_titular','cedula_titular']

class EtapaUno(Page):
    def is_displayed(self):
        return self.round_number==1

class Bienvenida(Page):
    def is_displayed(self):
        return self.round_number==1

class InformedConsent_1(Page):
    def is_displayed(self):
        return self.round_number==1

class InformedConsent_2(Page):
    form_model = 'player'
    form_fields = ['city','day','month','year','hour','minutes','ampm','name','signature','idcard','citycard']
    def is_displayed(self):
        return self.round_number==1

class instructionsInicio(Page):
    def is_displayed(self):
        return self.round_number==1

class Puntaje(Page):
    def is_displayed(self):
        return self.round_number==1

page_sequence = [
    Introduction,
    InformedConsent_1,
    InformedConsent_2,
    Bienvenida,
    EtapaUno,
    Puntaje,
    Survey,
    AfterSurvey,
    Survey2,
    WaitForAll,
    Offer,
    ExpectationsB,
    WaitForA,
    RoundResults,
    DespuesRondaUno,
    InstructionsLastRoundB,
    InstructionsLastRoundC,
    SocialNormsB,
    SocialNormsC,
    AllGroupsWaitPage,
    ResultsWaitPage,
    Results,
    Payment,
    Questionnaire_A,
    Questionnaire_B,
    Pagos_1,
    Pagos_2,
    FinalMessage
]