from otree.api import Currency as c, currency_range
from . import pages
from ._builtin import Bot
from .models import Constants
import random

#
#
class PlayerBot(Bot):
     def play_round(self):
         if self.round_number == 1 and self.player.id_in_group == 1:
             yield (pages.Experimenter, {'counterbalance_choice': 'group52'})
         if self.round_number == 1:
             yield (pages.Personal_ID, {'personal_ID': 'Test_ID'})
             yield (pages.Welcome)
             yield (pages.RulesPage01)
             yield (pages.RulesPage10)
             yield (pages.RulesPage11)
             yield (pages.RulesPage20)
             yield (pages.RulesPage21)
             yield (pages.RulesPage22)
             yield (pages.RulesPage30)
             yield (pages.RulesPage31)
             yield (pages.RulesPage40)
             yield (pages.RulesPage41)

             if self.player.condition == "honest":
                 yield (pages.ComprQuest_1, {'compr_quest_1': 1, 'compr_quest_2': 1, 'compr_quest_3': 1, 'compr_quest_4': 1, 'compr_quest_5': 1, 'compr_quest_6': 1, 'compr_quest_7': 2})
             else:
                 yield (pages.ComprQuest_1, {'compr_quest_1': 1, 'compr_quest_2': 1, 'compr_quest_3': 1, 'compr_quest_4': 1, 'compr_quest_5': 1, 'compr_quest_6': 1, 'compr_quest_7': 1})

             yield (pages.ComprQuest_2, {'compr_quest_8': 1, 'compr_quest_9': 5, 'compr_quest_10': 4, 'compr_quest_11': 5, 'compr_quest_12': 1, 'compr_quest_13': 1, 'compr_quest_14': 1})
             yield (pages.ComprQuest_3, {'compr_quest_15': 3, 'compr_quest_16': 3, 'compr_quest_17': 4, 'compr_quest_18': 1, 'compr_quest_19': 4, 'compr_quest_20': 3, 'compr_quest_21': 5})

             if self.player.leader == 'Leader':
                 yield (pages.Summary_Leader)
             else:
                 yield (pages.Summary_Fighter)

         if self.player.leader == 'Leader':
             if self.player.condition == "honest":
                 if self.player.role_player == "Attacker":
                    yield (pages.LeaderRole, {'leader_role': 'Attacker'})
                 else:
                    yield (pages.LeaderRole, {'leader_role': 'Defender'})
             else:
                 yield (pages.LeaderRole, {'leader_role': 'Attacker'})
         if self.player.leader == 'Leader':
             if self.round_number == 1:
                 yield (pages.Expectations, {'expectation_confl': 15})
             else:
                 yield (pages.Expectations, {'expectation_confl': 10})
         if self.player.leader == 'Fighter':
             if self.round_number == 1:
                 yield (pages.Contribute, {'contribution_confl': 0, 'contribution_keep': 20})
             else:
                 yield (pages.Contribute, {'contribution_confl': 10, 'contribution_keep': 10})
         yield (pages.Results)
         if self.round_number == 20:
             yield (pages.Endoftask)
             yield (pages.Questions_Cohesion, {'coh1_solidarity': 2, 'coh2_solidarity': 2, 'coh3_solidarity': 2, 'coh4_satisfaction': 2, 'coh5_satisfaction': 2, 'coh6_satisfaction': 2, 'coh7_satisfaction': 2})
             if self.player.leader == 'Fighter':
                 if self.player.condition == "honest":
                     yield (pages.Questions_Fighters_Honesty_hon, {'exp_RoleArounds': 10, 'exp_RoleBrounds': 10, 'reelect': 'Yes'})
                 else:
                     yield (pages.Questions_Fighters_Honesty_dec, {'exp_RoleArounds': 10, 'exp_RoleBrounds': 10, 'exp_signalhonest': 10, 'reelect': 'Yes'})
                 yield (pages.Questions_Scenario, {'ScenarionAA': 2, 'ScenarionAB': 2, 'ScenarionBA': 2, 'ScenarionBB': 2})
                 yield (pages.Questions_Fighters_Endorsement, {'increase_leaderendorse': 1, 'decrease_leaderendorse':9})
             if self.player.leader == 'Leader':
                 if self.player.condition == "honest":
                     yield (pages.Questions_Leader_Honesty_hon, {'exp_RoleArounds': 10, 'exp_RoleBrounds': 10})
                 else:
                     yield (pages.Questions_Leader_Honesty_dec, {'exp_RoleArounds': 10, 'exp_RoleBrounds': 10, 'exp_signalhonest': 10})
                 yield (pages.Questions_Scenario, {'ScenarionAA': 2, 'ScenarionAB': 2, 'ScenarionBA': 2, 'ScenarionBB': 2})
                 yield (pages.Questions_Leader_Endorsement)

    #         yield (pages.Earning)