from .models import *

class Consent_form(Page):
    form_model = 'player'
class Treatment_compassion(Page):
    form_model = 'player'
    def is_displayed(self):
        group = self.group
        player = self.player
        return player.id_in_group == 1
class Treatment_mindfulness(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', 'Nationality', 'Meditation_check', 'Follow_video']
class Instructions_Trust_Game(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,
        
        total_amount= 100 + player.send_amount * 0.5,
                   )
page_sequence = [Consent_form, Treatment_compassion, Treatment_mindfulness, Control, Demographics, Instructions_Trust_Game, Send, Results]