from .models import *

class Start(Page):
    form_model = 'player'
class BFI_1(Page):
    form_model = 'player'
    form_fields = ['b1', 'b2', 'b3', 'b4', 'b5', 'b6', 'b7', 'b8', 'b9', 'b10', 'b11', 'b12', 'b13', 'b14', 'b15', 'b16', 'b17', 'b18', 'b19', 'b20', 'b21', 'b22']
class BFI_2(Page):
    form_model = 'player'
    form_fields = ['b23', 'b24', 'b25', 'b26', 'b27', 'b28', 'b29', 'b30', 'b31', 'b32', 'b33', 'b34', 'b35', 'b36', 'b37', 'b38', 'b39', 'b40', 'b41', 'b42', 'b43', 'b44']
class SVO(Page):
    form_model = 'player'
    form_fields = ['s1', 's2', 's3', 's4', 's5', 's6', 's7', 's8', 's9']
class Demographics(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'degree', 'major', 'grade', 'course', 'university', 'work', 'hometown', 'income', 'pay', 'lastname', 'part1_report', 'part1_reason', 'part1_guilty', 'part3_guilty']
class Payoff(Page):
    form_model = 'player'
    def vars_for_template(self):
        player = self.player
        return set_payoff(player)
page_sequence = [Start, BFI_1, BFI_2, SVO, Demographics, Payoff]