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

doc = ''
class Constants(BaseConstants):
    name_in_url = 'trust_game'
    players_per_group = 2
    num_rounds = 1
    name = 'trust_game'
    endowment = c(10)
    multiplier = 3
class Subsession(BaseSubsession):
    pass
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
def sent_back_amount_max(group):
    return group.sent_amount * 3
class Group(BaseGroup):
    sent_amount = models.CurrencyField(label='How much do you want to send to Participant B?', max=Constants.endowment)
    sent_amount = sent_amount
    sent_back_amount_max = sent_back_amount_max
    sent_back_amount = models.CurrencyField(label='How much do you want to send back?', max=sent_back_amount_max)
    set_payoffs = set_payoffs
class Player(BasePlayer):
    validation = models.IntegerField(label='How many points do both players have?')
def custom_export(players):
    yield ['participant_code', 'id_in_group']
    for p in players:
        pp = p.participant
        yield [pp.code, p.id_in_group]