from otree.api import Currency as c, currency_range

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


class Belief_EFD(Page):
    form_model = 'player'
    form_fields = ['EFD1',
                   'EFD2',
                   'EFD3',
                   'EFD4',
                   ]

class Email(Page):
    form_model = 'player'
    form_fields = ['Email',
                   ]
    def is_displayed(self):
        return self.participant.vars.get('treatment_group') == 'treatment'

class End_Screen(Page):
    pass

page_sequence = [
    Belief_EFD,
    Email,
    End_Screen,
]