import random

from otree.api import *


"""
Sim for 'Frontrunner - Challenger' game. Players are matched randomly and must choose a political platform or reaction
to their opponent's political platform.
"""


class Constants(BaseConstants):
    name_in_url = 'frontrunner'
    players_per_group = 2
    num_rounds = 5
    instructions_template = 'frontrunner/instructions.html'
    matrix_template = 'frontrunner/matrix.html'
    row_role = 'Row'
    column_role = 'Column'


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    choice = models.StringField(widget=widgets.RadioSelect, label='')


# FUNCTIONS
def creating_session(subsession: Subsession):
    if subsession.round_number == 1:
        subsession.group_randomly()
    else:
        subsession.group_like_round(1)


def choice_choices(player: Player):
    if player.role == Constants.row_role:
        return ['Extreme', 'Moderate', 'Vague']
    else:
        return ['Challenge', 'Ignore', 'Praise']


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


def set_payoff(player: Player):
    payoff = {
        'Extreme': {'Challenge': [10, -10], 'Ignore': [-5, 5], 'Praise': [0, 0]},
        'Moderate': {'Challenge': [0, 0], 'Ignore': [7, -7], 'Praise': [1, -1]},
        'Vague': {'Challenge': [4, -4], 'Ignore': [3, -3], 'Praise': [2, -2]},
    }
    i = 0 if player.role == Constants.row_role else 1
    player.payoff = payoff[player.group.get_player_by_role('Row').choice][
        player.group.get_player_by_role('Column').choice
    ][i]


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


class Main(Page):
    form_model = 'player'
    form_fields = ['choice']


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):
        return {
            'player_payoff': int(player.payoff),
            'my_choice': player.choice.lower(),
            'opponent_choice': other_player(player).choice.lower(),
        }


class Final(Page):
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 5

    @staticmethod
    def vars_for_template(player: Player):
        opponent = other_player(player)
        my_total = int(player.participant.payoff)
        opponent_total = int(opponent.participant.payoff)
        return {'my_payoff': my_total, 'opponent_payoff': opponent_total}


page_sequence = [Introduction, Main, ResultsWaitPage, Results, Final]