import random
import itertools
import numpy
from numpy import random
from random import randrange, random, randint

from Introduction.config import *
from otree.api import *

from . import models

author = 'Your name here'


class Constants(BaseConstants):
    name_in_url = 'Aufgabe'
    players_per_group = None
    endowment = 10
    num_rounds = 1
    Aufgabe_template = 'Aufgabe/Aufgabe_template.html'
    Nachrichten_template = 'Aufgabe/Nachrichten_template.html'
    Nachrichten_2_template = 'Aufgabe/Nachrichten_template_2.html'


class Subsession(BaseSubsession):
    pass


def creating_session(subsession: Subsession) :
    colors = itertools.cycle(['tmt1', 'tmt2', 'tmt3', 'tmt4', 'tmt4'])
    for player in subsession.get_players():
        player.color = next(colors)

class Group(BaseGroup):
    pass

class Player(BasePlayer):
    belief = models.IntegerField(label="Die soeben gezeigten Beispiele entsprechen meiner <b>Wahrnehmung</b> von "
                                         "Versicherungsunternehmen", widget=widgets.RadioSelect,
                                 choices=[[1, "Stimme überhaupt nicht zu"], [2, "Stimme nicht zu"],
                                            [3, "Stimme weder zu noch lehne ich ab"], [4, "Stimme zu"],
                                            [5, "Stimme voll und ganz zu"]])
    experience = models.IntegerField(label="Die soeben gezeigten Beispiele entsprechen meinen <b>Erfahrungen</b> mit "
                                         "Versicherungsunternehmen", widget=widgets.RadioSelect,
                                     choices=[[1, "Stimme überhaupt nicht zu"], [2, "Stimme nicht zu"],
                                            [3, "Stimme weder zu noch lehne ich ab"], [4, "Stimme zu"],
                                            [5, "Stimme voll und ganz zu"]])
    color = models.StringField()

# ******************************************************************************************************************** #
# *** PAGES *** #
# ******************************************************************************************************************** #


class Aufgabe(Page):
    pass

class Nachrichten(Page):
    form_model = 'player'
    form_fields = ['belief',
                   'experience']
    @staticmethod
    def is_displayed(player: Player):
        return player.color == 'tmt3' or player.color == 'tmt4'

class Nachrichten_2(Page):
    form_model = 'player'
    form_fields = ['belief',
                   'experience']
    @staticmethod
    def is_displayed(player: Player):
        return player.color == 'tmt1' or player.color == 'tmt2'

class SorryFull(Page):
    pass

# ******************************************************************************************************************** #
# *** PAGE SEQUENCE *** #
# ******************************************************************************************************************** #

page_sequence = [
    Aufgabe, Nachrichten, Nachrichten_2
]