from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c, currency_range
)


author = 'Salim Nuhu'

doc = """
This is the CoP Experiment
"""

import random
class Constants(BaseConstants):

    name_in_url = 'Exam'
    players_per_group = 2
    num_rounds = 1
    endowment = c(100)
    multiplication_factor = 3
    shocks = [0, 3]
    #random_return = random.choice(shocks)


    #instructions_template = 'COP_Experiment/instructions.html'

class Subsession(BaseSubsession):
    def creating_session(self):
        self.group_randomly(fixed_id_in_group=True)
        if self.round_number == 1:
            random_return = random.choice(Constants.shocks)
            self.session.vars['random_return'] = random_return



class Group(BaseGroup):
    sent_amount = models.CurrencyField(
        min=0, max=Constants.endowment,
        label='''
        I would like to send P2
        ''',
    )

    #tripled = models.CurrencyField(c(3) * c(sent_amount)
    #)

    sent_back_amount = models.CurrencyField(
        min=c(0), max= Constants.endowment*Constants.multiplication_factor,
        label='''
        I would like to send back to P1
        ''',
    )






class Player(BasePlayer):
    def role(self):
        if self.id_in_group == 1:
            return 'Sender'
        if self.id_in_group == 2:
            return 'Receiver'