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

class PlayerBot(Bot):
    def play_round(self):
        if self.round_number == 1:
            if self.player.role == Constants.role_agent:
                yield pages.initial_agent, dict(first_bid_agent=0)
                yield pages.bid_agent_1_visible, dict(visible_cut=50,visible_bid=50)
                yield pages.bid_agent_1_secret, dict(secret_cut=100,secret_bid=100)
            elif self.player.role == Constants.role_principal:
                yield pages.initial_principal, dict(first_bid_principal=600)
                yield pages.bid_principal_1_visible, dict(visible_cut=550,visible_bid=550)
                yield pages.bid_principal_1_secret, dict(secret_cut=500,secret_bid=500)
        elif self.round_number == 2:
            if self.player.role == Constants.role_agent:
                yield pages.bid_agent_n_visible, dict(visible_cut=150,visible_bid=150)
                yield pages.bid_agent_n_secret, dict(secret_cut=200,secret_bid=200)
            elif self.player.role == Constants.role_principal:
                yield pages.bid_principal_n_visible, dict(visible_cut=450,visible_bid=450)
                yield pages.bid_principal_n_secret, dict(secret_cut=400,secret_bid=400)
        elif self.round_number == 3:
            if self.player.role == Constants.role_agent:
                yield pages.bid_agent_n_visible, dict(visible_cut=350,visible_bid=350)
                yield pages.bid_agent_n_secret, dict(secret_cut=400,secret_bid=400)
                expect(self.group.settlement, 300)
                yield pages.Result_solved
            elif self.player.role == Constants.role_principal:
                yield pages.bid_principal_n_visible, dict(visible_cut=250,visible_bid=250)
                yield pages.bid_principal_n_secret, dict(secret_cut=200,secret_bid=200)
                expect(self.group.settlement, 300)
                yield pages.Result_solved