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

import random
import re

author = 'Magnus Strobel'

doc = """
Your app description
"""


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


class Subsession(BaseSubsession):
    def creating_session(self):
        if self.round_number == 1:
            for p in self.get_players():
                assigned_treatment_group = random.choices(
                    population=['treatment', 'control'],
                    weights=[0.25, 0.75],
                    k=1)[0]
                p.participant.vars['treatment_group'] = assigned_treatment_group
                p.treatment_group = assigned_treatment_group
                
                p.participant.vars["treatment_reminder"] = "treatment/Treatment_reminder.html"


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    treatment_group = models.StringField()