from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c, currency_range
)
from otreeutils.surveys import create_player_model_for_survey
# Import modules
from ibb.modules.boredom_proneness import boredom_proneness
from ibb.modules.state_boredom import state_boredom
from ibb.modules.state_boredom_after import state_boredom_after
from ibb.modules.personal_info import personal_info
from ibb.modules.ibt import ibt
from ibb.modules.consent import consent
# Additional libraries
import random
import json


class Constants(BaseConstants):
    name_in_url = 'BoredomProneness'
    players_per_group = None
    tasks = ['ES01', 'ES02', 'ES03', 'ES04', 'ES05', 'ES06',
             'LS01', 'LS02', 'LS03', 'LS04', ]
    num_rounds = len(tasks)
    sale_pages = [1, 2, 3, 4, 5, 6]

class Subsession(BaseSubsession):
    def creating_session(self):
        for i, p in enumerate(self.get_players()):
            ''' The odd number players gets treatment 1 and,
                the evens get treatment 2. Treatment 1 will
                get shown the Peg Turning task.'''
            p.treatment = (i % 2) + 1

            ''' Each player will be assigned one of six extra
                shop pages at random. That page will show an
                ongoing sale! '''
            p.sale_page = random.choice(Constants.sale_pages)

            ''' Change page sequence. Instead of a single round with
                all pages, X rounds will be played with each round
                consisting of a single page. This allows us to control
                which page to show and randomize it on a per player
                basis. '''
            round_numbers = list(range(1, Constants.num_rounds+1))
            random.shuffle(round_numbers)
            p.participant.vars['round'] = dict(zip(Constants.tasks, round_numbers))

class Group(BaseGroup):
    pass

# define survey questions per page
# for each page define a page title and a list of questions
# the questions have a field name, a question text (input label), and a field type (model field class)
SURVEY_DEFINITIONS = (
    personal_info,
    boredom_proneness,
    state_boredom,
    ibt,
    consent,
    state_boredom_after,
)

OTHER_FIELDS = {
    'treatment': models.IntegerField(),
    'sale_page': models.IntegerField(), 
    'peg': models.IntegerField(),
    'list_01': models.IntegerField(),
    'list_02': models.IntegerField(), 
    'list_03': models.IntegerField(),
    'list_04': models.IntegerField(),
    'extra_01': models.IntegerField(),
    'extra_02': models.IntegerField(),
    'extra_03': models.IntegerField(),
    'extra_04': models.IntegerField(),
    'extra_05': models.IntegerField(),
    'extra_06': models.IntegerField(),
    'feedback': models.LongStringField(label="Thank you for taking part in the study! I would be very happy if you could leave me your feedback."),
}

# now dynamically create the Player class from the survey definitions
# we can also pass additional (non-survey) fields via `other_fields`
Player = create_player_model_for_survey('ibb.models', SURVEY_DEFINITIONS, other_fields=OTHER_FIELDS)