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

import numpy as np


author = 'Siva and Ashish'

doc = """
Your app description
"""


class Constants(BaseConstants):
	name_in_url = 'auction_nps'
	players_per_group = 2
	num_rounds = 20
	n_treats=4;	rounds_per_treat=5;	n_periods=3;
	# multipliers=np.array([1,1,1,1,1,1,2,2,2,2])
	multipliers=np.array([1,1,1])
	# l=10	#l<c_l<u-diff and c_l<c_h<u
	# u=25
	diff=5
	F=8; V=7;


class Subsession(BaseSubsession):
	treatment=models.IntegerField(choices=[0,1]) #B=0 and NB=1
	def creating_session(self):
		u=np.random.randint(low=20, high=50, size=1)
		ratio= 0.1* np.random.random_sample()+0.8
		if self.round_number in range(1, int(Constants.num_rounds/2)+1):
			self.treatment=0
		else:
			self.treatment=1

		if self.round_number in range(1, int(Constants.num_rounds/4)+1) or self.round_number in range(int(Constants.num_rounds/2)+1, int(3*Constants.num_rounds/4)+1):
			self.group_randomly(fixed_id_in_group=True)
		else:
			self.group_randomly(fixed_id_in_group=True)
			matrix=self.get_group_matrix()
			for group in matrix:
				group.reverse()
			self.set_group_matrix(matrix)
		for g in self.get_groups():
			# c_l = np.random.randint(low=Constants.l, high=Constants.u-Constants.diff, size=1)
			c_l=u*ratio
			for p in g.get_players():
				if p.role()=='low_cost':
					p.cost= np.around(c_l,1)
					p.cost_role='low_cost'
				else:
					# p.cost= np.random.randint(low=c_l+Constants.diff, high=Constants.u+1, size=1)
					p.cost= np.around(u,1)
					p.cost_role='high_cost'



class Group(BaseGroup):
	def call_auction(self):
		#Update results for group players
		p = self.get_players()
		i = p[0].participant.vars['current_i']
		if p[0].participant.vars['current_i']>Constants.n_periods:
			return
		bids=[getattr(player, 'bid'+str(i)) for player in p]
		winning_bid=min(bids)
		if bids[0]==bids[1]:
			temp = np.random.choice([True,False])
			results = np.array([temp, not temp])
		else:
			results=np.array(bids)==winning_bid
		opponent_bids=[bids[1], bids[0]]
		for idx, player in enumerate(p):
			exec("player.result"+str(i)+"= results[idx]")
			exec("player.opponent_bid"+str(i)+"= bids[(idx+1)%2]")
			if results[idx]:
				exec("player.profit"+str(i)+"= bids[idx]-player.cost")
				profit =bids[idx]-player.cost
			else:
				exec("player.profit"+str(i)+"= 0")
				profit = 0
			dummy = ['Lost', 'Won']
			profit=profit*Constants.multipliers[i-1]
			player.participant.vars['bids'].append(bids[idx])
			player.participant.vars['profits'].append(round(profit, 2))
			player.participant.vars['results'].append(dummy[results[idx]])
			player.participant.vars['opponent_bids'].append(opponent_bids[idx])
			player.participant.vars['winning_bids'].append(winning_bid)
			player.participant.vars['current_i']+=1

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

	cost_role=models.TextField()

	scaling=models.FloatField()

	cost = models.FloatField()	
	all_bids = models.TextField()
	all_profits = models.TextField()
	all_opponent_bids = models.TextField()
	all_results = models.TextField()

	bid1=models.FloatField(min=0.0)
	opponent_bid1=models.FloatField()
	result1=models.BooleanField()
	profit1=models.FloatField()

	bid2=models.FloatField()
	opponent_bid2=models.FloatField()
	result2=models.BooleanField()
	profit2=models.FloatField()

	bid3=models.FloatField()
	opponent_bid3=models.FloatField()
	result3=models.BooleanField()
	profit3=models.FloatField()

	bid4=models.FloatField()
	opponent_bid4=models.FloatField()
	result4=models.BooleanField()
	profit4=models.FloatField()

	bid5=models.FloatField()
	opponent_bid5=models.FloatField()
	result5=models.BooleanField()
	profit5=models.FloatField()

	bid6=models.FloatField()
	opponent_bid6=models.FloatField()
	result6=models.BooleanField()
	profit6=models.FloatField()

	bid7=models.FloatField()
	opponent_bid7=models.FloatField()
	result7=models.BooleanField()
	profit7=models.FloatField()

	bid8=models.FloatField()
	opponent_bid8=models.FloatField()
	result8=models.BooleanField()
	profit8=models.FloatField()

	bid9=models.FloatField()
	opponent_bid9=models.FloatField()
	result9=models.BooleanField()
	profit9=models.FloatField()

	bid10=models.FloatField()
	opponent_bid10=models.FloatField()
	result10=models.BooleanField()
	profit10=models.FloatField()