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


class Practice_Round(Page):
   def is_displayed(self):
    return self.round_number<=3

class Round(Page):
   def is_displayed(self):
    return self.round_number>=4


class Firm_Effort(Page):
    """This page is only for Firm
    to decide on Effort """

    form_model = 'group'
    form_fields = ['firm_effort']

    def is_displayed(self):
        return self.player.id_in_group == 1

class Student_Effort(Page):
    """This page is only for Student
    to decide on Effort """

    form_model = 'group'
    form_fields = ['student_effort']

    def is_displayed(self):
        return self.player.id_in_group == 2

class Wait(WaitPage):
        pass


class Firm_Price(Page):
        """This page is only for Firm
        to decide on Effort """

        form_model = 'group'
        form_fields = ['firm_price']

        def is_displayed(self):
            return self.player.id_in_group == 1


class Student_Buy(Page):
            """This page is only for Firm
            to decide on Effort """

            form_model = 'group'
            form_fields = ['student_buy']

            def is_displayed(self):
                return self.player.id_in_group == 2

class ResultsWaitPage(WaitPage):
            def after_all_players_arrive(self):
                    self.group.set_payoffs()

            body_text = "Waiting for result."

class Results(Page):
    pass



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



page_sequence = [
    Practice_Round,
    Round,
    Firm_Price,
    Wait,
    Firm_Effort,
    Student_Effort,
    Wait,
    Student_Buy,
    ResultsWaitPage,
    Results
]