from otree.api import Currency as c, currency_range, SubmissionMustFail, Submission, expect
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.Type_WillBeAssigned
            yield pages.Type_Assign
            yield pages.BeliefsWillStart
            yield pages.BeliefsInstructions
            yield pages.Beliefs, dict(beliefs_HH=10, beliefs_HL=20, beliefs_LH=25, beliefs_LL=0)
            yield pages.PGGInstructions
            yield pages.FirstRoundWillStart


        yield pages.Partner_Matched
        yield pages.Partner_Assigned
        yield pages.Contribute,  dict(contribution=25)
        yield pages.Results

        if self.round_number == 24:
            yield pages.Attention_Check_2_Page, dict(Attention_Check_2=200)