from otree.api import *

doc = """
Your app description
"""


class Constants(BaseConstants):
    name_in_url = 'inst'
    players_per_group = 2
    num_rounds = 1


class Subsession(BaseSubsession):
    pass


def creating_session(subsession):
    subsession.group_randomly()
    group_matrix = subsession.get_group_matrix()
    subsession.session.vars['group_matrix'] = group_matrix



    for player in subsession.get_players():
        # set treatment from session config, if not set before
        if player.participant.vars.get('treatment') is None:
            player.treatment = subsession.session.config['treatment']
            player.participant.vars['treatment'] = player.treatment
        else:
            player.treatment = player.participant.vars['treatment']

        # set types by group ids, if not grouped before
        if player.participant.vars.get('type') is None:
            if player.id_in_group == 1:
                player.type = "A"
            if player.id_in_group == 2:
                player.type = "B"
            player.participant.vars['type'] = player.type
        else:
            player.type = player.participant.vars['type']


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    treatment = models.StringField()
    type = models.StringField()


# PAGES
class General1(Page):
    pass


class General2(Page):
    pass


page_sequence = [General1, General2]