import random

from otree.api import *

author = 'Kailash Rajah'

doc = """
Your app description
"""


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


class Subsession(BaseSubsession):
    pass

class Group(BaseGroup):
    multiplication_factor = models.IntegerField(
        initial=3
    )
    sent_amount = models.CurrencyField(
        label="How much do you want to send to Player B?",
        min=cu(0),
        max=Constants.endowment
    )
    sent_back_amount = models.CurrencyField(
        label="How much do you want to send back?",
    )
    message = models.LongStringField(
        label = "What message would you like to send Participant B",
        blank = True
    )


#    def sent_back_amount_choices(self):
#       return currency_range(
#            cu(0),
#            self.sent_amount * Constants.multiplication_factor,
#            cu(1)
#        )
#

class Player(BasePlayer):
    pass

#Functions
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

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

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

class WaitforP1(WaitPage):
    pass

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

    def is_displayed(player):
        return player.id_in_group == 2

    def vars_for_template(player):
        group = player.group

        return dict(
            tripled_amount=group.sent_amount * group.multiplication_factor
        )

class MyPage(Page):
    pass

class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs

class Results(Page):
    pass


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