from otree.api import Currency as c, currency_range

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.Introduction)
            yield (pages.Lottery, {'lottery_choice': 4})
            yield (pages.Instructions, {'stdchoice': 2})
        submitted_answer = self.player.current_question()['choice1']
        yield (pages.Question, {'submitted_answer': submitted_answer})
        yield (pages.Probability, {'probability': 50})
        if self.round_number == Constants.num_rounds:
            yield (pages.BeliefsPrior1, {'prior_num_guess': 10, 'practice_guess1': 4})
            yield (pages.BeliefsPrior2, {'prior_prob_guess1': 5, 'prior_prob_guess2': 20, 'prior_prob_guess3': 50, 'prior_prob_guess4': 20, 'prior_prob_guess5': 5})
            yield (pages.Request, {'practice1': 2, 'practice2': 1})
            yield (pages.Request2, {'s0': False, 's1': False, 's2': False, 's3': False, 's4': False, 's5': True, 's6': True, 's7': True, 's8': True, 's9': True})
            yield (pages.BeliefsPost1, {'post_num_guess': 10})
            yield (pages.BeliefsPost2, {'post_prob_guess1': 5, 'post_prob_guess2': 20, 'post_prob_guess3': 50, 'post_prob_guess4': 20, 'post_prob_guess5': 5})
            yield (pages.Results)