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

doc = '\nThis is a standard 2-player trust game where the amount sent by player 1 gets\ntripled. The trust game was first proposed by\n<a href="http://econweb.ucsd.edu/~jandreon/Econ264/papers/Berg%20et%20al%20GEB%201995.pdf" target="_blank">\n    Berg, Dickhaut, and McCabe (1995)\n</a>.\n'
class Constants(BaseConstants):
	name_in_url = 'trust'
	num_rounds = 1
	players_per_group = None
	endowment = c(25)
	test_list = (0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
	instructions_template = 'trust/instructions.html'
class Subsession(BaseSubsession):
	pass
class Group(BaseGroup):
	sent_amount = models.CurrencyField(doc='Amount sent by P1', max=Constants.endowment, min=0)
	sent_back_amount = models.CurrencyField(doc='Amount sent back by P2', min=0)
	risk = models.IntegerField(initial=0, max=10)
	question2_a = models.IntegerField(max=10, min=0)
	question2_b = models.IntegerField(max=10, min=0)
	question2_c = models.IntegerField(max=10, min=0)
	def sent_back_amount_max(self):
		return self.sent_amount * Constants.multiplier
	def set_payoffs(self):
		p1 = self.get_player_by_id(1)
		p2 = self.get_player_by_id(2)
		p1.payoff = Constants.endowment - self.sent_amount + self.sent_back_amount
		p2.payoff = self.sent_amount * Constants.multiplier - self.sent_back_amount
class Player(BasePlayer):
	risk = models.IntegerField(label='Disposition à prendre des risques', max=4)
	def role(self):
		return {1: 'A', 2: 'B'}[self.id_in_group]