from .models import *

class Earnings(Page):
    form_model = 'player'
    def vars_for_template(self):
        participant = self.participant
        return dict(
        earnings=participant.payoff_plus_participation_fee()-6)
class Demographics(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'ethnicity', 'year_at_UVA', 'volunteer_hours', 'economics', 'resident']
class Survey_Questions(Page):
    form_model = 'player'
    form_fields = ['strategy', 'dishes', 'efficiency_equity', 'dictator_game']
page_sequence = [Earnings, Demographics, Survey_Questions]