from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c, currency_range
)

doc = ''


class Constants(BaseConstants):
    name_in_url = 'survey_bsb'
    players_per_group = None
    num_rounds = 1


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    first_field = models.IntegerField(choices=[[1, '1'], [2, '2'], [3, '3'], [4, '4'], [5, '5']],
                                      widget=widgets.RadioSelectHorizontal)

    second_field = models.IntegerField(choices=[[1, '1'], [2, '2'], [3, '3'], [4, '4'], [5, '5']],
                                       widget=widgets.RadioSelectHorizontal)

    third_field = models.IntegerField(choices=[[1, '1'], [2, '2'], [3, '3'], [4, '4'], [5, '5']],
                                      widget=widgets.RadioSelectHorizontal)

    departement = models.StringField()

    def get_dept(self):
        if self.participant.label == 'bsb_01':
            return "departement_01"
        else:
            return "departement_02"