from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c, currency_range
)


author = 'Žan Mlakar'

doc = """
Master Thesis Experiment Tryout #1
"""


class Constants(BaseConstants):
    name_in_url = 'Tryout1'
    players_per_group = 2
    num_rounds = 5

    same_choice = c(10)


class Subsession(BaseSubsession):
    def creating_session(self):
        print('in creating_session', self.round_number)


class Group(BaseGroup):
    def same_answer(self):
        print('same_answer')
        player1 = self.get_player_by_id(1)
        player2 = self.get_player_by_id(2)

        if player1.chosen_name == player2.chosen_name:
            player1.is_same = True
            player2.is_same = True
        else:
            player1.is_same = False
            player2.is_same = False
        for player in [player1, player2]:
            if player.is_same:
                player.payoff = Constants.same_choice
            else:
                player.payoff = c(0)


class Player(BasePlayer):
    chosen_name = models.StringField(
        choices=['John', 'Jack'],
        widget=widgets.RadioSelectHorizontal
    )

    is_same = models.BooleanField()