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

class Intro(Page):
    pass
#    def is_displayed(self):
#        return self.round_number == 1

class Instructions(Page):
    def vars_for_template(self):
        return {
            'heterogeous'
            'homogeneous'
        }
    pass
#    def is_displayed(self):
#        return self.round_number == 1

class Efficacy(Page):
    form_model = 'player'
    form_fields = ['E1', 'E2', 'E3', 'E4', 'E5']
        #add efficacy HET/CIT here ['E1', 'E2', 'E3', 'E4', 'E5', 'E6', 'E7', 'E8', 'E9', 'E10', 'E11', 'E12']
#    def is_displayed(self):
#        return self.round_number == 1

class Decision(Page):
    form_model = 'player'
    form_fields = ['invest']
#    def is_displayed(self):
#        return self.round_number == 1

class ResultsWaitPage(WaitPage):
    def after_all_players_arrive(self):
        self.group.set_payoffs()
#    def is_displayed(self):
#        return self.round_number == 1

class Results_Round(Page):
    def vars_for_template(self):
        return {
            'round_payoff': self.player.payoff,
            'num_investors': self.group.num_investors,
            'total_shares': self.group.sum_shares,
            'good_produced': self.group.good_is_produced()
        }
#    def is_displayed(self):
#        return self.round_number == 1

class End_Sum(Page):
        pass
#    def vars_for_template(self):
#        return {
#            'total_payoff': sum([p.payoff for p in self.player.in_all_rounds()]),
#        }
# is.displayed  def is_displayed(self):
#         return self.round_number == Constants.num_rounds

page_sequence = [
    Intro,
    Instructions,
    Efficacy,
    Decision,
    ResultsWaitPage,
    Results_Round,
    End_Sum,
]