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


author = 'Your name here'

doc = """
Your app description
"""


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


class Subsession(BaseSubsession):
    
    def creating_session(self):
        import random
        for p in self.get_players():
            p.treat = random.choice(['blue', 'red'])
            if p.treat == "blue":
                p.test = 1
            else:
                p.test = 2
            
        
    


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    treat = models.StringField()
    test = models.IntegerField(label = "Treatment State")