from otree.api import Bot, Currency as c
from . import pages
from .models import Constants
app = pages
cu = c

class PlayerBot(Bot):
    def play_round(self):
        if self.player.id_in_group == 1:
            yield pages.Lender_Page, dict(lending=True)
        if self.player.id_in_group == 2 and self.group.lending:
            yield pages.Borrower1, dict(borrower1=True, monitoring1=True)
        if self.player.id_in_group == 3 and self.group.lending:
            yield pages.Borrower2, dict(borrower2=True, monitoring2=True)
        yield pages.Results
        if self.player.round_number == Constants.num_rounds:
            yield pages.ResultsSummary