from otree.api import *


doc = """
Your app description
"""


class Constants(BaseConstants):
    name_in_url = 'my_trust'
    players_per_group = 2
    num_rounds = 1
    endowment = 10
    multiplication_factor = 3


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    role_A_payoff = models.IntegerField()
    role_B_payoff = models.IntegerField()
    sent_back_max = models.IntegerField()
    sent_amount = models.IntegerField(min=0, max=Constants.endowment,
        label="How much do you want to send to role B?"
    )
    sent_back_amount = models.IntegerField(min=0, max=sent_back_max,
        label="How much do you want to send back?"
    )


def sent_back_amount_max(group):
    return group.sent_back_max


def set_payoffs(group):
    group.role_A_payoff = Constants.endowment - group.sent_amount + group.sent_back_amount
    group.role_B_payoff = Constants.endowment + group.sent_amount * Constants.multiplication_factor - group.sent_back_amount


class Player(BasePlayer):
    student_ID = models.IntegerField(label="Please enter your student ID to proceed")


# PAGES
class ID_Entry(Page):
    form_model = 'player'
    form_fields = ['student_ID']


class Send(Page):
    form_model = 'group'
    form_fields = ['sent_amount']

    @staticmethod
    def is_displayed(player):
        return player.id_in_group == 1

    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        group = player.group
        group.sent_back_max = group.sent_amount * Constants.multiplication_factor


class WaitforP1(WaitPage):
    pass


class SendBack(Page):
    form_model = 'group'
    form_fields = ['sent_back_amount']

    @staticmethod
    def is_displayed(player):
        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):
    pass


page_sequence = [ID_Entry, Send, WaitforP1, SendBack, ResultsWaitPage, Results]