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 self.subsession.round_number == 1:
            yield (pages.AttentionCheck, {'attention_check': '1 2 3 4 5 6 7 8 9 0 1 2 3 4 5'})
            yield (pages.Welcome, {'prolific_id': 'a1'})
            yield (pages.Consent)
            yield (pages.Instructions)
            yield (pages.Example, {'answer':50,})
            yield (pages.Machine)
            yield (pages.SequenceEvents)
            a = random.choice([0,1])
            yield (pages.ComprehensionQns, {
                'qn_confidence1': 12,
                # 'qn_select': 1,
                'qn_draws': 1,
                'qn_balls': a,
            })
            if self.participant.vars['failed_comprehension']:
                yield (pages.FailedComprehension)
        if not self.participant.vars['failed_comprehension']:
            if self.subsession.round_number == 1:
                   yield (pages.BeginStudy)
            yield Submission(pages.Wait, check_html=False)
            a = random.randint(0, 100)
            yield (pages.Belief_new, {
                'answer': a,
               
            })
            yield (pages.Confidence, {
                'confidence': random.randint(0, 20),
            })