from otree.api import Currency as c, currency_range

from ._builtin import Page, WaitPage
from .models import Constants

class Priming(Page):
    def is_displayed(self):
        if self.session.config['treatment'] == 1:
            return True
        elif self.session.config['treatment'] == 2:
            return True
        elif self.session.config['treatment'] == 3:
            return False
        elif self.session.config['treatment'] == 4:
            return False

    form_model = 'player'
    form_fields = ['campusliving', 'roommate', 'floorsex', 'floorsexpref', 'environmentsex', 'environmentsexpref', 'prefreason1', 'prefreason2', 'prefreason3']

class NoPriming(Page):
    def is_displayed(self):
        if self.session.config['treatment'] == 1:
            return False
        elif self.session.config['treatment'] == 2:
            return False
        elif self.session.config['treatment'] == 3:
            return True
        elif self.session.config['treatment'] == 4:
            return True

    form_model = 'player'
    form_fields = ['cellphonepref', 'videopref', 'laptoppref', 'mobilesitepref', 'touchscrenpref', 'remotepref', 'prefreason1', 'prefreason2', 'prefreason3']

page_sequence = [Priming, NoPriming]