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

doc = '\nSimple trust game\n'
class Constants(BaseConstants):
    name_in_url = 'My_trust_game'
    players_per_group = 2
    num_rounds = 3
    endowment = c(10)
    multiplier = 3
    instructions_template = 'My_trust_game/instructions.html'
class Subsession(BaseSubsession):
    pass
def sent_back_amount_choices(group):
    return currency_range(c(0), group.sent_amount * Constants.multiplier, c(1))
def set_payoffs(group):
    p1 = group.get_player_by_id(1)
    p2 = group.get_player_by_id(2)
    
    p1.payoff = Constants.endowment - group.sent_amount + group.sent_back_amount
    p2.payoff = group.sent_amount * Constants.multiplier - group.sent_back_amount
class Group(BaseGroup):
    sent_amount = models.CurrencyField(doc='Amount sent by P1', label='Please enter an amount from 0 to 100', max=Constants.endowment, min=0)
    sent_back_amount = models.CurrencyField(doc='Amount sent back by P2', label='0')
    sent_back_amount_choices = sent_back_amount_choices
    set_payoffs = set_payoffs
class Player(BasePlayer):
    Name = models.StringField(blank=True, label='Name')
    Occupation = models.StringField(blank=True, label='Occupation/Field of study')
    Age = models.IntegerField(label='Age', max=98, min=0)
    Place = models.StringField(blank=True, label='Place')
    Gender = models.StringField(choices=[['Female', 'Female'], ['Male', 'Male'], ['Diverse', 'Diverse']], label='Gender')