from otree.api import *

doc = """
Your app description
"""


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


class Subsession(BaseSubsession):
    pass


def creating_session(subsession):
    import random
    for group in subsession.get_groups():
        group.multiplication_factor = random.choice([6, 3])
        group.round_incentivized = random.choice([1, 2, 3])


class Group(BaseGroup):
    multiplication_factor = models.IntegerField()
    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?"
    )
    round_incentivized = models.IntegerField()
    message = models.LongStringField(
        label="You may also send a message to participant B.",
        blank=True,
    )
    pass


def sent_back_amount_choices(group):
    return currency_range(
        0,
        group.sent_amount * group.multiplication_factor,
        1
    )


class Player(BasePlayer):
    final_payoff = models.CurrencyField()
    pass


# PAGES
class Send(Page):
    form_model = 'group'
    form_fields = ['sent_amount', 'message']

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

    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 * group.multiplication_factor
        )

    pass


class WaitForP1(WaitPage):
    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 * group.multiplication_factor - group.sent_back_amount


class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs
    pass


def set_final_payoffs(group):
    for player in group.get_players():
        player_incentivized = player.in_round(group.round_incentivized)
        player.final_payoff = player_incentivized.payoff


class FinalResultsWaitPage(WaitPage):
    @staticmethod
    def is_displayed(group):
        return group.round_number == 3
    after_all_players_arrive = set_final_payoffs
    pass

class Results(Page):
    pass


class PaymentPage(Page):
    @staticmethod
    def is_displayed(group):
        return group.round_number == 3
    set_final_payoffs
    pass

page_sequence = [
    Send,
    WaitForP1,
    SendBack,
    ResultsWaitPage,
    Results,
    FinalResultsWaitPage,
    PaymentPage
]