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

class Introduction(Page):
    pass


class Basics(Page):
    form_model = 'player'
    form_fields = ['state', 'state_years','age', 'education']

    def before_next_page(self):
        self.player.set_trueprop()


class Main1_NY(Page):
    form_model = 'player'
    form_fields = ['attention1']

    def is_displayed(self):
        return self.player.display1() and self.player.state == "NY"


class Main1_NJ(Page):
    form_model = 'player'
    form_fields = ['attention1']

    def is_displayed(self):
        return self.player.display1() and self.player.state == "NJ"


class Main1_ME(Page):
    form_model = 'player'
    form_fields = ['attention1']

    def is_displayed(self):
        return self.player.display1() and self.player.state == "ME"


class Main2(Page):
    form_model = 'player'
    form_fields = ['attention2','approve_ever', 'approve_month']

    def is_displayed(self):
        return self.player.display1() 


class Main3(Page):
    form_model = 'player'
    form_fields = ['donation']

    def is_displayed(self):
        return self.player.display1() and self.player.donation_q() 


class Wait(Page):
    timeout_seconds = 0.01

    def is_displayed(self):
        return self.player.display1() 

    def before_next_page(self):
        self.player.set_payoff()        


class Results(Page):
    pass


page_sequence = [Introduction,
                 Basics,
                 Main1_NY,
                 Main1_NJ,
                 Main1_ME,
                 Main2,
                 Main3,
                 Wait,
                 Results]