from .models import *

class Consent_form(Page):
    form_model = 'player'
class Treatment_c(Page):
    form_model = 'player'
    def is_displayed(self):
        group = self.group
        player = self.player
        return player.id_in_group == 1
class Treatment_m(Page):
    form_model = 'player'
    def is_displayed(self):
        group = self.group
        player = self.player
        return player.id_in_group == 2
class Control(Page):
    form_model = 'player'
    def is_displayed(self):
        group = self.group
        player = self.player
        return player.id_in_group == 3
class Demographics(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'education', 'Occupation', 'Nationality', 'Meditation_check', 'Follow_video', 'Video_content_check']
class Instructions(Page):
    form_model = 'player'
class Send(Page):
    form_model = 'player'
    form_fields = ['send_amount']
class Results(Page):
    form_model = 'player'
    def vars_for_template(self):
        player = self.player
        return dict(received_amount= player.send_amount * 1.5,
        
                    multiplied_amount= player.send_amount*3,
        
        total_amount= 100 + player.send_amount * 0.5,
                   )
page_sequence = [Consent_form, Treatment_c, Treatment_m, Control, Demographics, Instructions, Send, Results]