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


class Information(Page):
    def is_displayed(self):
        return self.round_number == 1

class Portfolio_Vignette(Page):
    def is_displayed(self):
        return self.round_number == 1
    form_model = 'player'
    form_fields = ['portfolio_own_choice','portfolio_other_choice_prob',
                'portfolio_best_choice','portfolio_max_profits']
    def vars_for_template(self):
        if self.player.default_treatment == 'bank':
            best_choice_insert = 'der Bankenverband'
            max_profits_insert = 'des Bankenverbandes'
        elif self.player.default_treatment == 'state':
            best_choice_insert = 'die staatliche Behörde'
            max_profits_insert = 'der staatlichen Behörde'
        return dict(
            best_choice_label='Die Vorauswahl wurde getroffen, weil {} denkt, dass das Musterportfolio für den durchschnittlichen Bürger die beste Wahl ist.'.format(best_choice_insert),
            max_profits_label='Die Vorauswahl wurde getroffen, um die Gewinne {} zu erhöhen.'.format(max_profits_insert)
        )

class Flight_Vignette_1(Page):
    def is_displayed(self):
        if self.player.order_vignettes[0] == 'A':
            return self.round_number == 1
        elif self.player.order_vignettes[1] == 'A':
            return self.round_number == 2
        elif self.player.order_vignettes[2] == 'A':
            return self.round_number == 3
    form_model = 'player'
    form_fields = ['flight_random']

class Flight_Vignette_2(Page):
    def is_displayed(self):
        if self.player.order_vignettes[0] == 'A':
            return self.round_number == 1
        elif self.player.order_vignettes[1] == 'A':
            return self.round_number == 2
        elif self.player.order_vignettes[2] == 'A':
            return self.round_number == 3
    form_model = 'player'
    form_fields = ['flight_best_choice','flight_max_profits']

class Insurance_Vignette(Page):
    def is_displayed(self):
        if self.player.order_vignettes[0] == 'B':
            return self.round_number == 1
        elif self.player.order_vignettes[1] == 'B':
            return self.round_number == 2
        elif self.player.order_vignettes[2] == 'B':
            return self.round_number == 3
    form_model = 'player'
    form_fields = ['insurance_random','insurance_best_choice','insurance_max_profits']

class Garbage_Vignette(Page):
    def is_displayed(self):
        if self.player.order_vignettes[0] == 'C':
            return self.round_number == 1
        elif self.player.order_vignettes[1] == 'C':
            return self.round_number == 2
        elif self.player.order_vignettes[2] == 'C':
            return self.round_number == 3
    form_model = 'player'
    form_fields = ['garbage_random','garbage_best_choice','garbage_max_profits']

class Financial_Literacy(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds
    form_model = 'player'
    form_fields = ['financial_literacy_1','financial_literacy_2','financial_literacy_3']

class Results(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds


page_sequence = [Information,
                Portfolio_Vignette,
                Flight_Vignette_1,
                Flight_Vignette_2,
                Insurance_Vignette,
                Garbage_Vignette,
                Financial_Literacy,
                Results
                ]