from otree.api import *
from random import shuffle

from prisoner import NextGameWaitPage


doc = """
This is a standard 2-player trust game where the amount sent by player 1 gets
tripled. The trust game was first proposed by
<a href="http://econweb.ucsd.edu/~jandreon/Econ264/papers/Berg%20et%20al%20GEB%201995.pdf" target="_blank">
    Berg, Dickhaut, and McCabe (1995)
</a>.
"""


class Constants(BaseConstants):
    name_in_url = 'trust'
    players_per_group = 2
    num_rounds = 2
    instructions_template = 'trust/instructions.html'
    # Initial amount allocated to each player
    endowment = cu(100)
    multiplier = 3


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    sent_amount = models.CurrencyField(
        min=0,
        max=Constants.endowment,
        doc="""Amount sent by P1""",
        label="I will send",
    )
    sent_back_amount = models.CurrencyField(
        doc="""Amount sent back by P2""", 
        label="I will send back",
        min=cu(0)
    )


class Player(BasePlayer):
    pass


# FUNCTIONS
def sent_back_amount_max(group: Group):
    # return Constants.endowment + group.sent_amount * Constants.multiplier
    return group.sent_amount * Constants.multiplier


def set_payoffs(group: 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 = Constants.endowment + group.sent_amount * Constants.multiplier - group.sent_back_amount

def creating_session(subsession: Subsession):
    subsession.group_randomly(fixed_id_in_group=True)
    
    if subsession.round_number == 2:
        mat = subsession.get_group_matrix()
        for i in range(len(mat)):
            mat[i] = mat[i][::-1] # reverse the roles
        subsession.set_group_matrix(mat)


# PAGES
class Introduction(Page):
    timeout_seconds = 240
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == 1


class Send(Page):
    """This page is only for P1
    P1 sends amount (all, some, or none) to P2
    This amount is tripled by experimenter,
    i.e if sent amount by P1 is 5, amount received by P2 is 15"""

    timeout_seconds = 150
    form_model = 'group'
    form_fields = ['sent_amount']

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

    @staticmethod
    def error_message(player: Player, values):
        if values['sent_amount'] > 100:
            return f'The value must be less than or equal to 100'


class SendBackWaitPage(WaitPage):
    pass


class SendBack(Page):
    """This page is only for P2
    P2 sends back some amount (of the tripled amount received) to P1"""

    form_model = 'group'
    form_fields = ['sent_back_amount']
    timeout_seconds = 150

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

    @staticmethod
    def vars_for_template(player: Player):
        group = player.group

        tripled_amount = group.sent_amount * Constants.multiplier
        return dict(
            tripled_amount=tripled_amount,
            total=tripled_amount+Constants.endowment
        )
    @staticmethod
    def error_message(player: Player, values):
        if values['sent_back_amount'] > player.group.sent_amount * Constants.multiplier:
            return f'The amount must be less than or equal to {player.group.sent_amount * Constants.multiplier}'


class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs


class Results(Page):
    """This page displays the earnings of each player"""
    timeout_seconds = 30
    @staticmethod
    def vars_for_template(player: Player):
        group = player.group

        return dict(tripled_amount=group.sent_amount * Constants.multiplier)


page_sequence = [
    Introduction,
    Send,
    SendBackWaitPage,
    SendBack,
    ResultsWaitPage,
    Results,
    NextGameWaitPage
]