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


class Portfolio_Table(Page):
	form_model = 'player'
	form_fields = ['choice_selectPortfolio', 'complex_selectPortfolio', 'all_teams', 'current_team']

	def before_next_page(self):

		player = self.player

		# Before the player reaches the next page, the utility of his portfolio selection is calculated
		for i in [1, 2, 3]:
			if player.choice_selectPortfolio == i:
				player.u_selectPortfolio = player.u_selectPortfolio + Constants.team_utilities[i-1]


page_sequence = [Portfolio_Table]