from otree.api import *
c = cu

doc = ''
class C(BaseConstants):
    NAME_IN_URL = 'Trust_game'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1
    ENDOWMENT = cu(10)
    MULTIPLICATION_FACTOR = 3
    PROLIFIC_ID = 0
class Subsession(BaseSubsession):
    pass
class Group(BaseGroup):
    sent_amount = models.CurrencyField(label='How much do you want to send by participant B?', max=C.ENDOWMENT)
    sent_back_amount = models.CurrencyField(label='How much do you want to send back?')
def sent_back_amount_choices(group: Group):
    return currency_range(0,group.sent_amount*C.MULTIPLICATION_FACTOR,1)
def set_payoffs(group: Group):
    p1=group.get_player_by_id(1)
    p2=group.get_player_by_id(2)
    p1.payoff=C.ENDOWMENT-group.sent_amount+group.sent_back_amount
    p2.payoff=group.sent_amount*C.MULTIPLICATION_FACTOR-group.sent_back_amount
class Player(BasePlayer):
    ID_prolific = models.LongStringField(label='What is your prolific ID ?')
    age = models.IntegerField(label='What is your age ?')
class Page_age(Page):
    form_model = 'player'
    form_fields = ['age']
class Send_page(Page):
    form_model = 'group'
    form_fields = ['sent_amount']
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group==1
class WAITFORP1(WaitPage):
    pass
class Send_back(Page):
    form_model = 'group'
    form_fields = ['sent_back_amount']
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group==2
    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(triple_amount=group.sent_amount*C.MULTIPLICATION_FACTOR)
class RESULTSWAITPAGE(WaitPage):
    after_all_players_arrive = set_payoffs
class RESULTS(Page):
    form_model = 'player'
class PROLIFIC_page(Page):
    form_model = 'player'
    form_fields = ['ID_prolific']
class Completion_code_page(Page):
    form_model = 'player'
page_sequence = [Page_age, Send_page, WAITFORP1, Send_back, RESULTSWAITPAGE, RESULTS, PROLIFIC_page, Completion_code_page]