from otree.api import Currency as c, currency_range

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

class screen2(Page):

    form_model = 'player'
    form_fields = ['state']



class screen1(Page):

    form_model = 'player'
    form_fields = ['agree']

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


class screen3(Page):

    def is_displayed(self):
        return self.player.treatment == 6



class Cscreen4(Page):

    def is_displayed(self):
        return self.player.treatment == 6


class screen5(Page):

    def is_displayed(self):
        return self.player.treatment == 6




class Cscreen6(Page):

    def is_displayed(self):
        return self.player.treatment == 6

    timeout_seconds = 150

    form_model = 'player'
    form_fields = ['guess4']

    def vars_for_template(self):

        return {
            'q1': self.player.actual1,
            'q2': self.player.actual2,
            'q3': self.player.actual3,
         }


class Cscreen7(Page):

    def is_displayed(self):
        return self.player.treatment == 6


    form_model = 'player'
    form_fields = ['invest4']

    def vars_for_template(self):

        return {
            'number': self.player.investcorona,
        }

class Cscreen8(Page):

    def is_displayed(self):
        return self.player.treatment == 6
    timeout_seconds = 150

    form_model = 'player'
    form_fields = ['guess5']

    def vars_for_template(self):

        return {
            'q1': self.player.actual1,
            'q2': self.player.actual2,
            'q3': self.player.actual3,
            'guess4': self.player.guess4,
         }

class Cscreen9(Page):

    def is_displayed(self):
        return self.player.treatment == 6


    form_model = 'player'
    form_fields = ['invest5']

    def vars_for_template(self):

        return {
            'number': self.player.investcorona,
        }


class Cscreen10(Page):

    def is_displayed(self):
        return self.player.treatment == 6

    timeout_seconds = 150

    form_model = 'player'
    form_fields = ['guess6']

    def vars_for_template(self):

        return {
            'q1': self.player.actual1,
            'q2': self.player.actual2,
            'q3': self.player.actual3,
            'guess4': self.player.guess4,
            'guess5': self.player.guess5,

        }

class Cscreen11(Page):

    def is_displayed(self):
        return self.player.treatment == 6


    form_model = 'player'
    form_fields = ['invest6']

    def vars_for_template(self):

        return {
            'number': self.player.investcorona,
        }

class Cscreen12(Page):

    def is_displayed(self):
        return self.player.treatment == 6

    timeout_seconds = 150

    form_model = 'player'
    form_fields = ['guess7']

    def vars_for_template(self):

        return {
            'q1': self.player.actual1,
            'q2': self.player.actual2,
            'q3': self.player.actual3,
            'guess4': self.player.guess4,
            'guess5': self.player.guess5,
            'guess6': self.player.guess6,

        }

class Cscreen13(Page):

    def is_displayed(self):
        return self.player.treatment == 6


    form_model = 'player'
    form_fields = ['invest7']

    def vars_for_template(self):

        return {
            'number': self.player.investcorona,
        }


class Cscreen14(Page):

    def is_displayed(self):
        return self.player.treatment == 6

    timeout_seconds = 150

    form_model = 'player'
    form_fields = ['guess8']

    def vars_for_template(self):

        return {
            'q1': self.player.actual1,
            'q2': self.player.actual2,
            'q3': self.player.actual3,
            'guess4': self.player.guess4,
            'guess5': self.player.guess5,
            'guess6': self.player.guess6,
            'guess7': self.player.guess7,
        }

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


class survey1(Page):

    def is_displayed(self):
        return self.player.treatment != 4

    form_model = 'player'
    form_fields = ['owntoday','ownpredicted']


class survey2(Page):

    def is_displayed(self):
        return self.player.treatment != 4

    form_model = 'player'
    form_fields = ['problosejob','probreduceincome','findnewjob']


class survey3(Page):

    def is_displayed(self):
        return self.player.treatment !=4

    form_model = 'player'
    form_fields = ['growth6','growth12','unemployment6','unemployment12','inflation6','inflation12','stock6','stock12']


class survey4(Page):

    def is_displayed(self):
        return self.player.treatment != 4

    form_model = 'player'
    form_fields = ['consumer1','consumer2','consumer3','consumer4','consumer5','consumer6','consumer7','consumer8','consumer9','consumer10']


class survey5(Page):


    form_model = 'player'
    form_fields = ['gender','marital','day','month','year','religion','caste','income','education','health','distancezone','risk']



class survey6(Page):

    def is_displayed(self):
        return self.player.treatment !=4

    form_model = 'player'
    form_fields = ['inapp_socialgather', 'inapp_work','inapp_grocery','inapp_violation','Ceffort','Cgovtrue','Cgovtbetter','Seffort','Sgovtrue','Sgovtbetter']




class complete(Page):

    def vars_for_template(self):

        return {
            'code': self.player.completion_code,

        }

class screeninvest(Page):


    form_model = 'player'
    form_fields = ['investcorona']



page_sequence = [ screen1,screen2,screen3, Cscreen4, screen5, Cscreen6,screeninvest,Cscreen7, Cscreen8, Cscreen9,
                  Cscreen10,Cscreen11,Cscreen12,Cscreen13, Cscreen14, survey1,survey2,survey3, survey4, survey5, survey6,complete]