from otree.api import Currency as c, currency_range
from . import models
from ._builtin import Page, WaitPage
from .models import Constants

class Demographics(Page):
    form_model = models.Player
    form_fields = ['age', 'gender', 'political','education','income']


class TrustInPeople(Page):
    form_model = models.Player
    form_fields = ['tip_{}'.format(i) for i in range(1,4)]

class GeneralTrust(Page):
    form_model = models.Player
    form_fields = ['gts_{}'.format(i) for i in range(1,7)]
    def vars_for_template(self):
        return {'gts_scales': range(1,6)}


class TrustScale(Page):
    form_model = models.Player
    form_fields = ['ts_{}'.format(i) for i in range(1, 6)]

    def vars_for_template(self):
        return {'ts_scales': range(1, 6)}


class CynicalHostility(Page):
    form_model = models.Player
    form_fields = ['ch_{}'.format(i) for i in range(1, 6)]
    def vars_for_template(self):
        return {'ch_scales': range(1,7)}


class RevengeScale(Page):
    form_model = models.Player
    form_fields = ['rs_{}'.format(i) for i in range(1,13)]

    def vars_for_template(self):
        return {'rs_scales': range(1,6)}

class AltruisticPersonalityScale(Page):
    form_model = models.Player
    form_fields = ['aps_{}'.format(i) for i in range(1,22)]

    def vars_for_template(self):
        return {'aps_scales': range(1,6)}

class Debrief(Page):
    form_model = models.Player
    form_fields = ['debrief_{}'.format(i) for i in range(1,5)]

page_sequence = [
    Demographics,
    TrustScale,
    GeneralTrust,
    # RevengeScale,
    AltruisticPersonalityScale,
    CynicalHostility,
    # TrustInPeople,
    Debrief
]