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


author = '@Eduard'

doc = """
Experiment: basket of balls
"""


class Constants(BaseConstants):
    name_in_url = 'basket'
    pay_basket_blue = 200
    pay_basket_yellow = 100
    num_balls = 10
    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():
                p.basketBlueLeft = random.choice([True, False])


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    numBallsBlue = models.IntegerField()
    numBallsYellow = models.IntegerField()
    basketBlueLeft = models.BooleanField(initial=0)