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

class PlayerBot(Bot):
    def play_round(self):
        if self.round_number == 1:
            yield pages.Start
        if self.round_number==1:
            yield pages.Introduction
        if self.round_number==1:
            yield pages.T1instructions
        if self.round_number==1:
            yield pages.PaymentInfo
        if self.player.id_in_group==1:
            yield pages.ContributeP1, dict(contribution=c(100))
        if self.player.id_in_group == 2:
            yield pages.ContributeP2, dict(contribution=c(100))
        if self.player.id_in_group==3:
            yield pages.ContributeP3, dict(contribution=c(100))
        if self.player.id_in_group==4:
            yield pages.ContributeP4, dict(contribution=c(100))
        if self.player.id_in_group==5:
            yield pages.ContributeP5, dict(contribution=c(100))
        if self.player.id_in_group==6:
            yield pages.ContributeP6, dict(contribution=c(100))
        yield pages.Results
        if self.round_number==Constants.num_rounds:
            yield pages.FinalResults, dict()