from otree.api import *
from . import models


class LifeOrientationTest(Page):
    form_model = 'player'
    form_fields = ['q1', 'q2', 'q3', 'q4', 'q5', 'q6', 'q7', 'q8', 'q9', 'q10']

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

        self.participant.vars['optimism_score'] = self.player.optimism_score
        self.participant.vars['optimism_category'] = self.player.optimism_category

page_sequence = [LifeOrientationTest]