import random

from otree.api import *


"""
Sim for 'Non-credible Threat Game'
"""


class Constants(BaseConstants):
    name_in_url = 'nct'
    players_per_group = 2
    num_rounds = 2
    instructions_template = 'nct/instructions.html'
    instructions_template2 = 'nct/instructions2.html'
    role = random.choice([1, 2])


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    decision = models.StringField(
        widget=widgets.RadioSelect, initial='', label="Please make your choice."
    )


# FUNCTIONS
def decision_choices(player: Player):
    if player.id_in_group == 1:
        if player.round_number == 1:
            return ['In', 'Out']
        else:
            return ['Up', 'Down']
    else:
        if player.round_number == 1:
            return ['Up', 'Down']
        else:
            return ['A', 'B']


def other_player(player: Player):
    return player.get_others_in_group()[0]


def set_payoff(player: Player):
    p1 = player.group.get_player_by_id(1)
    p2 = player.group.get_player_by_id(2)
    if player.round_number == 1:
        payoff = {
            'In': {'Up': [500, 100], 'Down': [-1000, -1000]},
            'Out': {'': [100, 600]},
        }
    else:
        payoff = {
            'Up': {'A': [50, 100], 'B': [-50, -50]},
            'Down': {'A': [100, 50], 'B': [0, -100]},
        }
    player.payoff = payoff[p1.decision][p2.decision][player.id_in_group - 1]


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


class Stage(Page):
    pass


class P1Decision(Page):
    form_model = 'player'
    form_fields = ['decision']

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


class P2WaitPage(WaitPage):
    title_text = ''
    body_text = 'You are Player 2. Please wait for Player 1 to to make their choice.'

    @staticmethod
    def is_displayed(player: Player):
        return player.id_in_group == 2


class P2Decision(Page):
    form_model = 'player'
    form_fields = ['decision']

    @staticmethod
    def is_displayed(player: Player):
        if player.id_in_group == 2:
            return other_player(player).decision != 'Out'

    @staticmethod
    def vars_for_template(player: Player):
        opponent = other_player(player)
        return {'opponent_decision': opponent.decision}


class ResultsWaitPage(WaitPage):
    @staticmethod
    def after_all_players_arrive(group: Group):
        for p in group.get_players():
            set_payoff(p)


class Results(Page):
    @staticmethod
    def vars_for_template(player: Player):
        opponent = other_player(player)
        return {
            'opponent': opponent,
            'my_payoff': int(player.payoff),
            'opponent_payoff': int(opponent.payoff),
        }


page_sequence = [
    Introduction,
    Stage,
    P1Decision,
    P2WaitPage,
    P2Decision,
    ResultsWaitPage,
    Results,
]