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


class PlayerBot(Bot):

    def play_round(self):
        if self.subsession.round_number == 1:
            if self.participant.vars['failed_comprehension']:
                yield (pages.FailedComprehension)
        
        if not self.participant.vars['failed_comprehension']:
            yield Submission(pages.Wait, check_html=False)
            if self.subsession.round_number == 1 and (self.player.is_load or self.player.is_complex_prob):
                yield (pages.ElicitListExample)
            
            if self.player.is_load:
                yield (pages.ElicitListLoad, {
                    'switching_point': random.randint(0, self.player.lottery_amount), 'load_sum': 1,
                })
            else:
                yield (pages.ElicitList, {
                    'switching_point': random.randint(0, self.player.lottery_amount),
                })
            yield (pages.Valuation, {
                'confidence': random.randint(0, 20),
            })
            if self.player.is_load:
                yield (pages.LoadSumEntry, {'load_guess': 1})
#            if self.player.timed_out:
#                yield (pages.Timeout)