import random

from otree.api import *


"""
Sim for 'Repeated Game of Random Length'
 """


class Constants(BaseConstants):
    name_in_url = 'repeatedFlip'
    players_per_group = 2
    num_rounds = 99
    instructions_template = 'repeatedFlip/instructions.html'


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    new_round = models.BooleanField()


class Player(BasePlayer):
    choice = models.IntegerField(
        choices=[1, 2], widget=widgets.RadioSelect, label='Please make your choice.'
    )


# FUNCTIONS
def creating_session(subsession: Subsession):
    for g in subsession.get_groups():
        g.coin_flip()
        print(g.new_round)
        if g.new_round:
            subsession.group_randomly()


def coin_flip(group: Group):
    group.new_round = random.random() >= 0.50


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


def set_payoff(player: Player):
    payoff = {1: {1: 32, 2: 12}, 2: {1: 50, 2: 25}}
    player.payoff = payoff[player.choice][other_player(player).choice]


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


class Rematch(Page):
    @staticmethod
    def is_displayed(player: Player):
        return not player.group.new_round


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