from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer
)

import csv


author = 'Eveline Vandewal'

doc = """
Part 1
"""


class Constants(BaseConstants):
    name_in_url = 'GMTA2_Part1'
    players_per_group = None

    with open('GMTA2_Part1/statements.csv') as statements_file:
        statements = list(csv.DictReader(statements_file))

    num_rounds = len(statements)


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    statement_id = models.IntegerField()
    statement = models.StringField()

    submitted_answer = models.IntegerField(
        choices=[
            [1, 'Agree'],
            [2, 'Disagree']
        ],
        widget=widgets.RadioSelect
    )
    submitted_importance = models.IntegerField()

    def current_statement(self):

        return self.participant.vars['randomized_statements'][self.round_number - 1]