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

import random
import itertools
from random import shuffle
# </standard imports>

author = 'Dan Way'

doc = """
Cost Allocation
"""


class Constants(BaseConstants):
    name_in_url = 'bsw2_2'
    players_per_group = 2
    num_rounds = 1


class Subsession(BaseSubsession):
    def creating_session(self):

        if 'condition' in self.session.config:

            for g in self.get_groups():
                #for p in g.get_players():
                p1 = g.get_player_by_id(1)
                p2 = g.get_player_by_id(2)
                #...they'll be rematched later anyway
                p1.participant.vars['condition'] = self.session.config['condition']
                p2.participant.vars['condition'] = self.session.config['condition']



class Group(BaseGroup):
    pass


class Player(BasePlayer):

    condition = models.IntegerField()

    informed_consent = models.CharField(initial=None,
                                        choices=[('Yes', 'Yes'),
                                                 ('No', 'No')],
                                        verbose_name='',
                                        widget=widgets.RadioSelect())

    kc1 = models.CharField(initial=None,
                           choices=['True', 'False'],
                           verbose_name='',
                           widget=widgets.RadioSelect())

    kc2 = models.CharField(initial=None,
                           choices=['True', 'False'],
                           verbose_name='',
                           widget=widgets.RadioSelect())

    kc3 = models.CharField(initial=None,
                           choices=[('a', 'a. One randomly selected period'),
                                    ('b', 'b. All periods combined')],
                           verbose_name='',
                           widget=widgets.RadioSelect())

    kc4 = models.CharField(initial=None,
                           choices=['True', 'False'],
                           verbose_name='',
                           widget=widgets.RadioSelect())

    kc5 = models.CharField(initial=None,
                           choices=[('a', 'a. Costs are allocated based on the proportion of total firm revenue generated by each division.'),
                                    ('b', 'b. Costs are allocated equally between the two divisions, no matter what.'),
                                    ('c', 'c. Costs are allocated by a random computer-generated draw.')],
                           verbose_name='',
                           widget=widgets.RadioSelect())

    worker_id = models.CharField(initial='e')

    mturk_dupe = models.IntegerField()