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

# Pages are responsible for retrieving and passing back data from models to templates and vice versa.
# If you need to show something to a participant or to get his/her input, you need to indicate this in pages.py

#If you want to display something other than text (e.g. a variable) you need to use the function
#vars_for_template and make it return a dictionary. The index of the dictionary can then be used to display it on the page with {{ index }}.
# it is key that you indicate from which model you return a variable.
class Welcome(Page):

    def vars_for_template(self):
        return {'pool': Constants.pool,
                'players': Constants.players_per_group,
                'factor': Constants.efficiency_factor,
                'max': Constants.max,
                'treatment': self.subsession.treatment,
                'tipping_point': Constants.tipping_point*100}

# I split the Pages for the comprehension tests since the structure looks nicer. Does not have a practical meaning.
# For each Question and Answer pair i created a new page. You can decide if you want to show the page by the
# id _displayed line.

## a form field is something the participant can interact with. Since we defined it on a player level, we must specify it in the form_model section.


class Test_Control(Page):
    form_model = 'player'
    form_fields = ['test_control']

    def is_displayed(self):
        return self.subsession.treatment == 0

class Results_Control(Page):
    def vars_for_template(self):
        return {'test_control': self.player.test_control}

    def is_displayed(self):
        return self.subsession.treatment == 0


class Test1(Page):
    form_model = 'player'
    form_fields = ['test1']

    def is_displayed(self):
        return self.subsession.treatment == 1

class Test2(Page):
    form_model = 'player'
    form_fields = ['test2']

    def is_displayed(self):
        return self.subsession.treatment == 1

class Results_Test1(Page):
    def vars_for_template(self):
        return {'test1': self.player.test1}

    def is_displayed(self):
        return self.subsession.treatment == 1

class Results_Test2(Page):
    def vars_for_template(self):
        return {'test2': self.player.test2}

    def is_displayed(self):
        return self.subsession.treatment == 1


# Now we create a page for the player to decide what to take.
class Take(Page):
    # a form field is something the participant can interact with. Here we will asked them to pick how much they want to take
    form_model = 'player'
    form_fields = ['take']
    def vars_for_template(self):
        return {'max': Constants.max
        }

class ResultsWaitPage(WaitPage):
    # We use after_all_players_arrive to make sure we only start our calculation after every participant made their choice.
    # self.group.set_payoff and set_breakdown runs the functions we defined on the models page. It is important that we run
    # set_breakdown first, since it is an input for the payoff.

    def after_all_players_arrive(self):
        self.group.set_breakdown()
        self.group.set_payoffs()



class Results(Page):
    def vars_for_template(self):
        # here the dict() is used to convert our list to a dictionary. dict() and {} are equivalent, but use a different notation. Please be aware.

        return dict(
            payoff = self.player.payoff,
            take = self.player.take,
            total_points_left = self.group.total_points_left,
            pool_mult = self.group.total_points_left * Constants.efficiency_factor,
            breakdown = self.group.breakdown,
            treatment = self.subsession.treatment,
            share = self.group.resource_share,
            completion_code=self.player.completion_code
        )


# here we indicate in which sequence we want the pages to the played. You can repeat pages as well.
page_sequence = [Welcome,
                 Test_Control,
                 Results_Control,
                 Test1,
                 Results_Test1,
                 Test2,
                 Results_Test2,
                 Take,
                 ResultsWaitPage,
                 Results]