from otree.api import *
import itertools
import random
doc = """
Your app description
"""


class C(BaseConstants):
    NAME_IN_URL = 'trust_tutorial'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1

    endowment = cu(10)
    #multiplication_factor = 3

    #completion_code = random.randint(1000000, 9999999)


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    multiplication_factor = models.IntegerField()
    sent_amount = models.CurrencyField(
        label='How much do you want to send to participant B?',
        #        min=cu(0),
        #        max=C.endowment
    )
    sent_back_amount = models.CurrencyField(
        label='How much do you want to send back to participant A?'
    )


class Player(BasePlayer):
    # completion_code = models.IntegerField()
    pass


# Functions
def creating_session(subsession):
    treatment = itertools.cycle([3, 4])
    for group in subsession.get_groups():
        group.multiplication_factor = next(treatment)
        # '''
        for pNum in range(1, C.PLAYERS_PER_GROUP+1):
            p = group.get_player_by_id(pNum)
            p.participant.label = random.randint(1000000, 9999999)
        # '''


def sent_amount_choices(group):  # dynamic form field validation - see documentation
    return currency_range(
        cu(0),
        C.endowment,
        cu(0.1)
    )


def sent_back_amount_choices(group):  # dynamic form field validation - see documentation
    return currency_range(
        cu(0),
        group.sent_amount * group.multiplication_factor,
        cu(0.1)
    )


def set_payoffs(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 * group.multiplication_factor - group.sent_back_amount


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

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


class WaitForP1(WaitPage):
    pass


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

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

    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(
            tripled_amount=group.sent_amount * group.multiplication_factor
        )


class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs


class Results(Page):
    pass


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