from . import pages
from ._builtin import Bot
from otree.api import Submission
import random
from .models import Constants


class PlayerBot(Bot):

    def play_round(self):
        if not self.participant.vars['failed_comprehension']:
            yield Submission(pages.Wait, check_html=False)
            a = random.randint(0, 100)
            if self.subsession.round_number == 1 and self.player.is_complex:
                yield (pages.BeginStudy)
                #yield (pages.TimerInfo, {'qn_timer': True})
            if self.player.is_load:
                yield (pages.Belief_new, {
                    'answer': a, 'load_sum': 1,
                })
            else:
                yield (pages.Belief_new, {
                    'answer': random.randint(0, 100),
                })
            yield (pages.Confidence, {
                'confidence': random.randint(0, 20),
            })
            if self.player.is_load:
                yield (pages.LoadSumEntry, {
                    'load_guess': 1,
            })
            if self.player.timed_out:
                yield (pages.Timeout)