from otree.api import *
import random
from collections import defaultdict

doc = """
ASPD
"""

class C(BaseConstants):
    NAME_IN_URL = 'ASPD'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 18
    payoff_R1 = [350, 350, 350, 350, 350, 350, 350, 350, 350]
    payoff_S1 = [150, 150, 150, 150, 150, 150, 150, 150, 150]
    payoff_T1 = [600, 600, 600, 600, 360, 360, 360, 600, 600]
    payoff_D1 = [320, 320, 320, 320, 320, 160, 320, 160, 160]
    payoff_R2 = [400, 400, 400, 400, 400, 400, 400, 400, 400]
    payoff_S2 = [180, 180, 50, 50, 50, 50, 50, 50, 180]
    payoff_T2 = [500, 500, 500, 500, 500, 500, 500, 500, 500]
    payoff_D2 = [200, 380, 200, 380, 380, 200, 200, 200, 200]

class Subsession(BaseSubsession):
    pass

class Group(BaseGroup):
    task1_number = models.IntegerField()
    task2_number = models.IntegerField()
    selected_round = models.IntegerField()
    selected_task_be = models.IntegerField()
    selected_round_be = models.IntegerField()

class Player(BasePlayer):
    num_failed_attempts = models.IntegerField(initial=0)
    failed_too_many = models.BooleanField(initial=False)
    quiz1 = models.IntegerField(label='If the First Mover chose A, then the Seconder Mover observed the choice and decides A. Then the First Mover gets:')
    quiz2 = models.IntegerField(label='If the First Mover chose A, then the Seconder Mover observed the choice and decides A. Then the Second Mover gets:')
    quiz3 = models.IntegerField(label='If the First Mover chose A, then the Seconder Mover observed the choice and decides B. Then the First Mover gets:')
    quiz4 = models.IntegerField(label='If the First Mover chose A, then the Seconder Mover observed the choice and decides B. Then the Second Mover gets:')
    quiz5 = models.BooleanField(label="Does the First Mover know the choice of Second Mover prior to taking her decision?")
    quiz6 = models.BooleanField(label="Does the Second Mover know the choice of First Mover prior to taking her decision?")

    choice_1st =  models.BooleanField(
        choices=[
           [True, 'Top'],
           [False, 'Down'],
        ]
    )

    choice_2nd_Top =  models.BooleanField(
        choices=[
           [True, 'Left'],
           [False, 'Right'],
        ]
    )

    choice_2nd_Down =  models.BooleanField(
        choices=[
           [True, 'Left'],
           [False, 'Right'],
        ]
    )

    task1_number = models.IntegerField()
    task2_number = models.IntegerField()
    selected_round = models.IntegerField()
    selected_task_be = models.IntegerField()
    selected_round_be = models.IntegerField()

def creating_session(subsession: Subsession):
    if subsession.round_number == 1:
        for g in subsession.get_groups():
            game_numbers = [0, 1, 2, 3, 4, 5, 6, 7, 8]
            random.shuffle(game_numbers)
            k=0
            for i in range(9):
                g.in_round(i+1).task1_number = game_numbers[i]
                k=k+1
                for p in g.get_players():
                    p.in_round(i+1).task1_number = g.in_round(i+1).task1_number
            for i in range(9):
                g.in_round(i+10).task2_number = game_numbers[i]
                k=k+1
                for p in g.get_players():
                    p.in_round(i+10).task2_number = g.in_round(i+10).task2_number


        for g in subsession.get_groups():
            random_round = random.randint(1, 9)
            g.in_round(18).selected_round = random_round
            for p in g.get_players():
                p.in_round(18).selected_round = g.in_round(18).selected_round

        for g in subsession.get_groups():
            task = [0, 1, 2, 3, 4]
            random_task = random.choice(task)
            g.in_round(18).selected_task_be = random_task
            for p in g.get_players():
                p.in_round(18).selected_task_be = g.in_round(18).selected_task_be

            for i in range(1, 10):
                if g.in_round(i).task1_number == g.in_round(18).selected_task_be:
                    g.in_round(18).selected_round_be = i
                    break

class ASPD_GamePage_1st(Page):
    form_model = 'player'
    form_fields = ['choice_1st']

    @staticmethod
    def is_displayed(player: Player):
        return player.round_number <= 9

    @staticmethod
    def vars_for_template(player):
        task1_number = player.task1_number
        return dict(
            R1 = C.payoff_R1[task1_number],
            S1 = C.payoff_S1[task1_number],
            T1 = C.payoff_T1[task1_number],
            D1 = C.payoff_D1[task1_number],
            R2 = C.payoff_R2[task1_number],
            S2 = C.payoff_S2[task1_number],
            T2 = C.payoff_T2[task1_number],
            D2 = C.payoff_D2[task1_number]
            )

class ASPD_GamePage_2nd(Page):
    form_model = 'player'
    form_fields = ['choice_2nd_Down', 'choice_2nd_Top']

    @staticmethod
    def is_displayed(player: Player):
        return player.round_number >= 10

    @staticmethod
    def vars_for_template(player):
        task2_number = player.task2_number
        return dict(
            R1=C.payoff_R1[task2_number],
            S1=C.payoff_S1[task2_number],
            T1=C.payoff_T1[task2_number],
            D1=C.payoff_D1[task2_number],
            R2=C.payoff_R2[task2_number],
            S2=C.payoff_S2[task2_number],
            T2=C.payoff_T2[task2_number],
            D2=C.payoff_D2[task2_number]
            )

class ASPD_Instructions(Page):
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1

class ASPD_Comprehension_Test(Page):
    form_model = 'player'
    form_fields = ['quiz1', 'quiz2', 'quiz3', 'quiz4', 'quiz5', 'quiz6']

    @staticmethod
    def error_message(player: Player, values):
        solutions = dict(quiz1=350, quiz2=400, quiz3=200, quiz4=500, quiz5=False, quiz6=True)
        errors = {name: 'Wrong' for name in solutions if values[name] != solutions[name]}
        if errors:
            player.num_failed_attempts += 1
            if player.num_failed_attempts >= 100:
                player.failed_too_many = True
            else:
                return errors

    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1

class Failed(Page):
    @staticmethod
    def is_displayed(player: Player):
        return player.failed_too_many

class Wait(Page):
    def is_displayed(player:Player):
        return player.round_number==C.NUM_ROUNDS


class ResultsWaitPage(WaitPage):
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == C.NUM_ROUNDS

    @staticmethod
    def after_all_players_arrive(group: Group):
        selected_round = group.selected_round
        selected_round_be = group.selected_round_be
        selected_task = group.selected_task_be
        group_in_selected_round = group.in_round(selected_round)
        selected_payment = group_in_selected_round.task1_number
        player_lists = group.get_players()
        player_1 = player_lists[0]
        player_2 = player_lists[1]
        player_1_in_selected_round_be_1st = player_1.in_round(selected_round_be)
        player_1_in_selected_round_be_2nd = player_1.in_round(selected_round_be+9)
        player_2_in_selected_round_be_1st = player_2.in_round(selected_round_be)
        player_2_in_selected_round_be_2nd = player_2.in_round(selected_round_be + 9)
        player_1_in_selected_round = player_1.in_round(selected_round)
        player_2_in_selected_round = player_2.in_round(selected_round+9)
        if player_1_in_selected_round.choice_1st:
            if player_2_in_selected_round.choice_2nd_Top:
                player_1.payoff = C.payoff_R1[selected_payment]
                player_2.payoff = C.payoff_R2[selected_payment]
            else:
                player_1.payoff = C.payoff_S1[selected_payment]
                player_2.payoff = C.payoff_T2[selected_payment]
        else:
            if player_2_in_selected_round.choice_2nd_Down:
                player_1.payoff = C.payoff_T1[selected_payment]
                player_2.payoff = C.payoff_S2[selected_payment]
            else:
                player_1.payoff = C.payoff_D1[selected_payment]
                player_2.payoff = C.payoff_D2[selected_payment]
        player_1.participant.vars[__name__] = [str(player_1.payoff), 'First Mover', str(selected_round), str(player_2_in_selected_round_be_1st.choice_1st), str(player_2_in_selected_round_be_2nd.choice_2nd_Top), str(player_2_in_selected_round_be_2nd.choice_2nd_Down), str(selected_task)]
        player_2.participant.vars[__name__] = [str(player_2.payoff), 'Second Mover', str(selected_round+9), str(player_1_in_selected_round_be_1st.choice_1st), str(player_1_in_selected_round_be_2nd.choice_2nd_Top), str(player_1_in_selected_round_be_2nd.choice_2nd_Down), str(selected_task)]





page_sequence = [ASPD_Instructions, ASPD_Comprehension_Test, ASPD_GamePage_1st, ASPD_GamePage_2nd, ResultsWaitPage, Wait]