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

import random

author = 'Aikerim O'

doc = """
Dice guessing game with control and treatment groups.
"""


class Constants(BaseConstants):
    name_in_url = 'rolling_dice'
    players_per_group = None
    num_rounds = 20

    instructions_template = 'rolling_dice/Instructions.html'

    max_allowed_guess = 6
    min_allowed_guess = 1

    win = c(100)  # payment if win
    lose = c(0)  # payment if lose
    participation_fee = c(2000)


class Subsession(BaseSubsession):
    def creating_session(self):
        self.group_randomly(fixed_id_in_group=True)
        import itertools
        group_c = itertools.cycle(['treatment', 'control'])
        if self.round_number == 1:
            for player in self.get_players():
                player.participant.vars['group'] = next(group_c)




class Group(BaseGroup):
    pass


class Player(BasePlayer):
    def role(self):
        if self.participant.vars['group'] == 'treatment':
            return 'treatment'
        else:
            return 'control'

    dice_roll = models.IntegerField()
    guess = models.IntegerField(
        min=Constants.min_allowed_guess,
        max=Constants.max_allowed_guess,
        doc="Guess value of participant"
    )

    is_winner = models.BooleanField()

    def set_payoff(self):
        print("inside set_payoff")
        if self.guess == self.dice_roll:
            self.is_winner = True
        else:
            self.is_winner = False

        if self.is_winner != True:
            self.payoff = Constants.lose
        else:
            self.payoff = Constants.win

    def convert(self):
        return self.participant.payoff.to_real_world_currency(self.session)