from otree.api import models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer, Currency

class Constants(BaseConstants):
    name_in_url = 'prisoner_dilemma'
    players_per_group = 2
    num_rounds = 1

class Subsession(BaseSubsession):
    pass

class Group(BaseGroup):
    pass

class Player(BasePlayer):
    decision = models.StringField(
        choices=[['Cooperate', 'Cooperate'], ['Defect', 'Defect']],
        widget=widgets.RadioSelect,
        label='Please make a decision:'
    )
    player_payoff = models.CurrencyField()

    def other_player(self):
        return self.get_others_in_group()[0]

    def set_payoff(self):
        other_player = self.other_player()
        if self.decision == 'Cooperate' and other_player.decision == 'Cooperate':
            self.player_payoff = Currency(3)
        elif self.decision == 'Cooperate' and other_player.decision == 'Defect':
            self.player_payoff = Currency(0)
        elif self.decision == 'Defect' and other_player.decision == 'Cooperate':
            self.player_payoff = Currency(5)
        else:
            self.player_payoff = Currency(1)