import Egalitarian1_Proportional2 as pages
from . import *
c = cu

class PlayerBot(Bot):
    def play_round(self):
        yield pages.Consent
        yield pages.Overview
        yield pages.Task1
        if self.player.id_in_group<4:
            yield pages.InstructionIndividualTrustGroupSender
        if self.player.id_in_group>3:
            yield pages.InstructionIndividualTrustGroupReceiver
        if self.player.id_in_group < 4:
            yield pages.IndividualTrustGroupSend, dict(
                individual_send_group=C.INITIAL_MONEY,
            )
        if self.player.id_in_group>3:
            yield pages.IndividualTrustGroupSendBack, dict(
                return_to_individual1=0,
                return_to_individual2=0,
                return_to_individual3=0,
            )
        if self.player.id_in_group==1:
            yield pages.IndividualGuess_p1, dict(guess_return_P1=0)
        if self.player.id_in_group==2:
            yield pages.IndividualGuess_p2, dict(guess_return_P2=0)
        if self.player.id_in_group==3:
            yield pages.IndividualGuess_p3, dict(guess_return_P3=0)
        yield pages.Task2
        yield pages.FillerTask, dict(Raven1=1)
        yield pages.Task3
        if self.player.id_in_group<4:
            yield pages.InstructionGroupTrustGroupSender
        if self.player.id_in_group>3:
            yield pages.InstructionGroupTrustGroupReceiver
        if self.player.id_in_group<4:
            yield pages.GroupTrustGroupSend, dict(group_send_group=C.INITIAL_MONEY)
        if self.player.id_in_group>3:
            yield pages.GroupTrustGroupSendBack, dict(group_return_group=0)
        if self.player.id_in_group<4:
            yield pages.GroupGuess, dict(guess_return_group=0)
        yield pages.DemographicData, dict(
            Age=0,
            Gender="xyz",
            Major="xyz",
            English="xyz",
        )
        if self.player.id_in_group==1:
            yield pages.ResultPageSender1
        if self.player.id_in_group==2:
            yield pages.ResultPageSender2
        if self.player.id_in_group==3:
            yield pages.ResultPageSender3
        if self.player.id_in_group>3:
            yield pages.ResultPageReceiver