from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants




#-----------------------------------------------------------------------------------
# voting treatment page:
class SetupWaitPage(WaitPage):
	wait_for_all_groups = True
	def after_all_players_arrive(self):
		if self.subsession.round_number == 1:
			self.subsession.set_paying_round()
		players = self.subsession.get_players()
		self.subsession.set_order()
		for p in players: 
			p.set_MyPrefernces()
		groups = self.subsession.get_groups()
		for g in groups:
			g.set_ordering()
			g.eliminate_alternatives()

class Voting(Page):
	form_model = 'player'
	form_fields = ['vote']
	def vars_for_template(self):
		profile = self.player.MyPreferences
		temp = [0 for x in range(0,4)]
		temp[0] = Constants.preferences[self.player.group.Ordering][profile][0]
		temp[1] = Constants.preferences[self.player.group.Ordering][profile][1]
		temp[2] = Constants.preferences[self.player.group.Ordering][profile][2]
		temp[3] = Constants.preferences[self.player.group.Ordering][profile][3]

		return dict(
			preference_profiles = Constants.preferences[self.player.group.Ordering],
			my_number = self.player.id_in_group,
			my_preferences = temp,
			my_profile = profile,
			numeric_options = [self.group.Option1, self.group.Option2],
			options = [Constants.alternatives[self.group.Option1-1],Constants.alternatives[self.group.Option2-1]],
			eliminated = [Constants.alternatives[self.group.Eliminated1-1],Constants.alternatives[self.group.Eliminated2-1]],
			round_number = self.player.subsession.round_number,
			num_rounds = Constants.num_rounds
			)

class ResultsWaitPage(WaitPage):
	wait_for_all_groups = False
	def after_all_players_arrive(self):
		self.group.set_results()

class Results(Page):
	def vars_for_template(self):
		if self.player.subsession.round_number == Constants.num_rounds:
			p = self.player.in_round(self.player.subsession.paying_round)
			self.player.participant.vars['treatment_earnings'] = p.earnings

		temp1 = [0 for x in range(0,4)]
		profile = self.player.MyPreferences
		temp1[0] = Constants.preferences[self.player.group.Ordering][profile][0]
		temp1[1] = Constants.preferences[self.player.group.Ordering][profile][1]
		temp1[2] = Constants.preferences[self.player.group.Ordering][profile][2]
		temp1[3] = Constants.preferences[self.player.group.Ordering][profile][3]


		return dict(
			my_preferences = temp1,
			preference_profiles = Constants.preferences[self.player.group.Ordering],
			my_number = self.player.id_in_group,
			my_profile = profile,
			collective_choice = Constants.alternatives[self.player.group.Collective_Choice],
			numeric_collective_choice = self.player.group.Collective_Choice,
			earnings = temp1[self.player.group.Collective_Choice],
			round_number = self.player.subsession.round_number,
			num_rounds = Constants.num_rounds
			)

page_sequence = [
				# voting treatment
				SetupWaitPage,
				Voting,
				ResultsWaitPage,
				Results,
				]