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



class Results(Page):
	form = 'player'
	def vars_for_template(self):
		# bonus1
		self.player.num_for_bonus = self.participant.vars['num_for_bonus']
		self.player.picked_ball_color = self.participant.vars['ball_color']
		self.player.bonus1 = c(self.participant.vars['question_bonus'])
		option = self.participant.vars['option']
		# get the question payoff table
		img_src = 'urn{}.png'.format(self.player.num_for_bonus)

		# bonus 2
		total_assets = self.participant.vars['total_assets']
		points_aftertax = self.participant.vars['aftertax_all']
		self.player.bonus2 = c(points_aftertax/100)

		# total payoff
		self.player.payoff = self.player.bonus1 + self.player.bonus2


		self.player.participant_vars_dump = str(self.participant.vars)

		return dict(
			number = self.player.num_for_bonus,
			bonus1 = self.player.bonus1,
			ball_color = self.player.picked_ball_color,

			total_assets = total_assets,
			points_aftertax = points_aftertax,
			bonus2 = self.player.bonus2,
			bonus  = self.player.payoff,
			option = option,
			img_src = img_src)


class Completion_code(Page):
	form ='player'
	def vars_for_template(self):
		self.participant.code
		return dict(code = self.participant.code)


page_sequence = [Results, Completion_code]