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


class PlayerBot(Bot):

    def play_round(self):
        if self.subsession.round_number == 1:
            yield (views.Introduction)
        if self.player.id_in_group == 1:
            yield (views.Offer, {'principal_wage_offer': random.randint(0, 9)})
        if self.player.id_in_group == 2:
            yield (views.WorkEffort, {'agent_effort_0': random.randint(0, 9), 'agent_effort_1': random.randint(0, 9), 'agent_effort_2': random.randint(0, 9),
                                      'agent_effort_3': random.randint(0, 9), 'agent_effort_4': random.randint(0, 9), 'agent_effort_5': random.randint(0, 9),
                                      'agent_effort_6': random.randint(0, 9), 'agent_effort_7': random.randint(0, 9), 'agent_effort_8': random.randint(0, 9),
                                      'agent_effort_9': random.randint(0, 9)})



        #yield (views.ResultsWaitPage)
        yield (views.Results)