from otree.api import *
c = cu

doc = ''
class Constants(BaseConstants):
    players_per_group = 2
    num_rounds = 1
    name_in_url = 'my_trust'
    endowment = cu(10)
    multiplication_factor = 3
class Subsession(BaseSubsession):
    pass
def sent_back_amount_choices(group):
    return currency_range(cu(0), group.sent_amount * Constants.multiplication_factor, cu(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.multiplication_factor - group.sent_back_amount
class Group(BaseGroup):
    sent_amount = models.CurrencyField(label='How much do you want to send to participant B?')
    sent_back_amount = models.CurrencyField(label='How much do you want to send back?')
class Player(BasePlayer):
    sent_amount = models.CurrencyField()
    sent_back_amount = models.CurrencyField()
class Send(Page):
    form_model = 'group'
    form_fields = ['sent_amount']
    @staticmethod
    def is_displayed(player):
        group = player.group
        return player.id_in_group == 1
class WaitForP1(WaitPage):
    pass
class SendBack(Page):
    form_model = 'group'
    form_fields = ['sent_back_amount']
    @staticmethod
    def is_displayed(player):
        group = player.group
        return player.id_in_group == 2
    @staticmethod
    def vars_for_template(player):
        group = player.group
        return dict( tripled_amount=group.sent_amount * Constants.multiplication_factor )
class ResultsWaitPage(WaitPage):
    after_all_players_arrive = 'set_payoffs'
class Results(Page):
    form_model = 'player'
page_sequence = [Send, WaitForP1, SendBack, ResultsWaitPage, Results]