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

import random

author = 'Your name here'

doc = """
Your app description
"""


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


class Subsession(BaseSubsession):
    def creating_session(self):
        players = self.get_players()
        first_treatment = random.randint(1, 2)

        for p in players:
            if p.id_in_group == 1:
                p.treatment_group = first_treatment
            else:
                p.treatment_group = 3 - first_treatment

class Group(BaseGroup):
    pass


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

    choices = models.IntegerField(
        choices = [
            [1, 'Lottery 1'],
            [2, 'Lottery 2'],
        ],
        widget=widgets.RadioSelect
        )
    lottery_outcome = models.CurrencyField()
    probability_of_the_rare_event = models.IntegerField()
    X = models.CurrencyField()