from otree.api import ( models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer, Currency as c, currency_range, ) import random import csv import numpy doc = """ Part 2 (b) Series of one-shot "Prisoner's Dilemma" - 10 Rounds """ SUPERGROUP_NUM_ERR = 'Wrong number of players per supergroup' class Constants(BaseConstants): name_in_url = 'part2_b_I' players_per_group = 2 players_per_supergroup = 4 assert players_per_supergroup % players_per_group == 0, \ SUPERGROUP_NUM_ERR # To test: num_rounds = 1 num_rounds = 10 supergroup_threshold1 = 4 supergroup_threshold2 = 8 supergroup_threshold3 = 12 # payoff if 1 player defects and the other cooperates""", betray_payoff = c(100) betrayed_payoff = c(10) # payoff if both players cooperate or both defect both_cooperate_payoff = c(73) both_defect_payoff = c(43) def slice_list(input): ppg = Constants.players_per_group output = [input[i:i+ppg] for i in range(0, len(input), ppg)] for o in output: assert len(o) == ppg, SUPERGROUP_NUM_ERR return output class Subsession(BaseSubsession): def creating_session(self): if self.round_number == 1: assert len(self.get_players()) % \ Constants.players_per_supergroup == 0, \ SUPERGROUP_NUM_ERR import itertools random_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16] random.shuffle(random_list) random_ids = itertools.cycle(random_list) for p in self.get_players(): p.random_id = next(random_ids) for g in self.get_groups(): for p in g.get_players(): if p.random_id <= Constants.supergroup_threshold1: p.participant.vars['supergroup_task1'] = int(1) elif p.random_id > Constants.supergroup_threshold1 and p.random_id <= Constants.supergroup_threshold2: p.participant.vars['supergroup_task1'] = int(2) elif p.random_id > Constants.supergroup_threshold2 and p.random_id <= Constants.supergroup_threshold3: p.participant.vars['supergroup_task1'] = int(3) else: p.participant.vars['supergroup_task1'] = int(4) p.participant.vars['id_random'] = int(p.random_id) subgroups_task1 = set([p.vars['supergroup_task1'] for p in self.session.get_participants()]) new_matrix = [] for s in subgroups_task1: A = [p for p in self.get_players() if p.participant.vars['supergroup_task1'] == s] random.shuffle(A) sliced_supergroup = slice_list(A) new_matrix.extend(sliced_supergroup) print(new_matrix) self.set_group_matrix(new_matrix) # with open('strategic_draft0/matching_matrix.csv', 'w') as csvfile: # writer = csv.writer(csvfile) # writer.writerows(new_matrix) # with open('strategic_draft0/matching_matrix2.csv', 'w+') as my_csv: # csvWriter = csv.writer(my_csv, delimiter=',') # csvWriter.writerows(new_matrix) #numpy.savetxt('strategic_draft0/matching_matrix3.csv', new_matrix, delimiter=",") #if self.round_number == 1: # assign a random number to participants [1,N] #for p in self.get_players(): #p.participant.vars['id_random'] = random.choice([1,2,3,4,5,6,7,8]) class Group(BaseGroup): def set_payoffs(self): for p in self.get_players(): p.set_payoff() class Player(BasePlayer): decision = models.IntegerField( choices=[[1, 'X'], [0, 'Y']], doc="""This player's decision""", widget=widgets.RadioSelect, ) decision_guess = models.IntegerField(choices=[[0, '0'], [1, '1'], [2, '2'], [3, '3'], [4, '4']], min=0, max=4, widget=widgets.RadioSelectHorizontal, label="" ) # If group-size=6players switch to: #decision_guess = models.IntegerField(choices=[[1, '0%-10%'], [2, '11%-20%'], [3, '21%-30%'], # [4, '31%-40%'], [5, '41%-50%'], [5, '41%-50%'], # [6, '51%-60%'],[7, '61%-70%'], [8, '71%-80%'], [9, '81%-90%'], [10, '91%-100%']], # min=1, max=10, # widget=widgets.RadioSelectHorizontal, #) supergroup_task1 = models.LongStringField() random_id = models.IntegerField() decision_type = models.StringField() decision_partner = models.IntegerField() decision_task1_r1 = models.IntegerField() decision_task1_r2 = models.IntegerField() decision_task1_r3 = models.IntegerField() decision_task1_r4 = models.IntegerField() decision_task1_r5 = models.IntegerField() decision_task1_r6 = models.IntegerField() decision_task1_r7 = models.IntegerField() decision_task1_r8 = models.IntegerField() decision_task1_r9 = models.IntegerField() decision_task1_r10 = models.IntegerField() decision_task1_type_r1 = models.StringField() decision_task1_type_r2 = models.StringField() decision_task1_type_r3 = models.StringField() decision_task1_type_r4 = models.StringField() decision_task1_type_r5 = models.StringField() decision_task1_type_r6 = models.StringField() decision_task1_type_r7 = models.StringField() decision_task1_type_r8 = models.StringField() decision_task1_type_r9 = models.StringField() decision_task1_type_r10 = models.StringField() decision_guess_task1_r1 = models.LongStringField() decision_guess_task1_r5 = models.LongStringField() decision_guess_task1_r10 = models.LongStringField() payoff_int = models.IntegerField() # cooprate_task1_r1 = models.IntegerField() # cooprate_task1_r2 = models.IntegerField() # cooprate_task1_r3 = models.IntegerField() # cooprate_task1_r4 = models.IntegerField() # cooprate_task1_r5 = models.IntegerField() # cooprate_task1_r6 = models.IntegerField() # cooprate_task1_r7 = models.IntegerField() # cooprate_task1_r8 = models.IntegerField() # cooprate_task1_r9 = models.IntegerField() # cooprate_task1_r10 = models.IntegerField() total_coop_task1_r1 = models.IntegerField() total_coop_task1_r2 = models.IntegerField() total_coop_task1_r3 = models.IntegerField() total_coop_task1_r4 = models.IntegerField() total_coop_task1_r5 = models.IntegerField() total_coop_task1_r6 = models.IntegerField() total_coop_task1_r7 = models.IntegerField() total_coop_task1_r8 = models.IntegerField() total_coop_task1_r9 = models.IntegerField() total_coop_task1_r10 = models.IntegerField() payoff_task_r1 = models.CurrencyField() payoff_task_r2 = models.CurrencyField() payoff_task_r3 = models.CurrencyField() payoff_task_r4 = models.CurrencyField() payoff_task_r5 = models.CurrencyField() payoff_task_r6 = models.CurrencyField() payoff_task_r7 = models.CurrencyField() payoff_task_r8 = models.CurrencyField() payoff_task_r9 = models.CurrencyField() payoff_task_r10 = models.CurrencyField() total_coop = models.IntegerField() total_coop_s1 = models.IntegerField() total_coop_s2 = models.IntegerField() total_coop_s3 = models.IntegerField() total_coop_s4 = models.IntegerField() coop_rate_s1 = models.IntegerField() coop_rate_s2 = models.IntegerField() coop_rate_s3 = models.IntegerField() coop_rate_s4 = models.IntegerField() def other_player(self): return self.get_others_in_group()[0] def set_payoff(self): payoff_matrix = dict( X=dict( X=Constants.both_cooperate_payoff, Y=Constants.betrayed_payoff, ), Y=dict( X=Constants.betray_payoff, Y=Constants.both_defect_payoff ), ) self.payoff = payoff_matrix[self.decision_type][self.other_player().decision_type] self.decision_partner = self.other_player().decision self.supergroup_task1 = str(self.participant.vars['supergroup_task1']) self.random_id = int(self.participant.vars['id_random']) if self.decision == 1 and self.other_player().decision == 1: self.payoff_int = int(Constants.both_cooperate_payoff) elif self.decision == 0 and self.other_player().decision == 0: self.payoff_int = int(Constants.both_defect_payoff) elif self.decision == 1 and self.other_player().decision == 0: self.payoff_int = int(Constants.betrayed_payoff) else: self.payoff_int = int(Constants.betray_payoff) if self.round_number == 1: self.payoff_task_r1 = self.in_round(self.round_number).payoff if self.round_number == 2: self.payoff_task_r2 = self.in_round(self.round_number).payoff if self.round_number == 3: self.payoff_task_r3 = self.in_round(self.round_number).payoff if self.round_number == 4: self.payoff_task_r4 = self.in_round(self.round_number).payoff if self.round_number == 5: self.payoff_task_r5 = self.in_round(self.round_number).payoff if self.round_number == 6: self.payoff_task_r6 = self.in_round(self.round_number).payoff if self.round_number == 7: self.payoff_task_r7 = self.in_round(self.round_number).payoff if self.round_number == 8: self.payoff_task_r8 = self.in_round(self.round_number).payoff if self.round_number == 9: self.payoff_task_r9 = self.in_round(self.round_number).payoff if self.round_number == 10: self.payoff_task_r10 = self.in_round(self.round_number).payoff def role(self): if self.participant.vars['supergroup_task1'] == int(1): return '1' elif self.participant.vars['supergroup_task1'] == int(2): return '2' elif self.participant.vars['supergroup_task1'] == int(3): return '3' else: return '4'