from otree.api import *
c = cu

doc = "\nEach player decides if to free ride or to volunteer from which all will\nbenefit.\nSee: Diekmann, A. (1985). Volunteer's dilemma. Journal of Conflict\nResolution, 605-610.\n"
class C(BaseConstants):
    NAME_IN_URL = 'Practice'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 2
    NUM_OTHER_PLAYERS = 1
    REWARD_NONINVESTOR = 20
    REWARD_FAIL = 10
    INSTRUCTIONS_TEMPLATE = 'Practice/instructions.html'
class Subsession(BaseSubsession):
    investor_payoff = models.FloatField()
def creating_session(subsession: Subsession):
    session = subsession.session
    import random
    subsession.group_randomly()
    subsession.investor_payoff = random.randint(9,20)
    print(subsession.get_group_matrix()) 
class Group(BaseGroup):
    has_volunteer = models.BooleanField(initial=False)
class Player(BasePlayer):
    is_volunteer = models.BooleanField(initial=False)
    volunteer_id = models.IntegerField()
    has_volunteer = models.BooleanField(initial=False)
    NC_Taste = models.IntegerField(choices=[[1, '1. Very unlikely                             '], [2, '2. Unlikely                             '], [3, '3. Not sure                             '], [4, '4. Likely                             '], [5, '5. Very likely         ']], label='Admitting that your tastes are different from those of your friends. ', widget=widgets.RadioSelectHorizontal)
    NC_Argue = models.IntegerField(choices=[[1, '1. Very unlikely'], [2, '2. Unlikely '], [3, '3. Not sure'], [4, '4. Likely'], [5, '5. Very likely']], label='Arguing with a friend who has a very different opinion on an issue. ', widget=widgets.RadioSelectHorizontal)
    NC_Speak = models.IntegerField(choices=[[1, '1. Very unlikely'], [2, '2. Unlikely'], [3, '3. Not sure'], [4, '4. Likely'], [5, '5. Very likely']], label='Speaking your mind about an unpopular issue at a social occasion.\u2028', widget=widgets.RadioSelectHorizontal)
    investor_payoff = models.FloatField()
def live_method(player: Player, data):
    session = player.session
    subsession = player.subsession
    group = player.group
    player.investor_payoff = subsession.investor_payoff
    if group.has_volunteer:
        return
    
    if data.get('volunteer'):
        group.has_volunteer = True
        for p in player.get_others_in_group():
            p.is_volunteer = False
        player.is_volunteer = True
    
        # broadcast to the group that the game is finished.
        return {0: dict(finished=True)}
def get_partners(player: Player):
    group = player.group
    return player.get_others_in_group() 
def set_payoffs(player: Player):
    session = player.session
    subsession = player.subsession
    group = player.group
    player.investor_payoff = subsession.investor_payoff
    if group.has_volunteer:
        player.has_volunteer = True
        if player.is_volunteer == True:
            player.payoff = subsession.investor_payoff
        else:
            player.payoff = C.REWARD_NONINVESTOR
    else:
        player.payoff = C.REWARD_FAIL
        for p in player.get_others_in_group():
            p.payoff = C.REWARD_FAIL 
    
def get_all_rounds(player: Player):
    all_rounds = player.in_all_rounds()
    return all_rounds 
class DisplayInfo(Page):
    form_model = 'player'
    @staticmethod
    def vars_for_template(player: Player):
        partners = get_partners(player) 
        payoff = set_payoffs(player) 
        return dict(partners = partners,
                   payoff = payoff)
class InfoWait(WaitPage):
    wait_for_all_groups = True
class Decision(Page):
    form_model = 'player'
    timeout_seconds = 60
    timer_text = 'Time left: '
    live_method = 'live_method'
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        group = player.group
        return not group.has_volunteer
    @staticmethod
    def vars_for_template(player: Player):
        partners = get_partners(player) 
        payoff = set_payoffs(player) 
        return dict(partners = partners,
                   payoff = payoff)
class Results_Succeed(Page):
    form_model = 'group'
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return group.has_volunteer
    @staticmethod
    def vars_for_template(player: Player):
        payoff = set_payoffs(player) 
        all_rounds = get_all_rounds(player)
        return dict(payoff = payoff,
                   all_rounds = all_rounds)
class Results_Fail(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return not group.has_volunteer
    @staticmethod
    def vars_for_template(player: Player):
        payoff = set_payoffs(player) 
        all_rounds = get_all_rounds(player)
        return dict(payoff = payoff,
                   all_rounds = all_rounds)
class DecisionWait(WaitPage):
    wait_for_all_groups = True
class Investment_Finished(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        session = player.session
        subsession = player.subsession
        return player.subsession.round_number >= 2
class PracticeWait(WaitPage):
    wait_for_all_groups = True
page_sequence = [DisplayInfo, InfoWait, Decision, Results_Succeed, Results_Fail, DecisionWait, Investment_Finished, PracticeWait]