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


class PlayerBot(Bot):
    def play_round(self):
        yield Submission(pages.StartQuestions, check_html=False)

        if self.player.participant.vars['role'] == 'defender' and not self.player.participant.vars['inactive']:

            yield pages.BeliefsEE, dict(ee=randint(0, 100))
            yield pages.BeliefsPNB, dict(pnb=choice([0, 1]))
            yield Submission(pages.BeliefsNE, timeout_happened=True)  # dict(ne=-99) #dict(ne=randint(0, 100))
            yield pages.BeliefsHNB, dict(hnb=choice([0, 1]))
            # yield pages.AppropBEH_u, dict(beh_u=choice([1, 2, 3, 4]))
            # yield pages.AppropBEH_d, dict(beh_d=choice([1, 2, 3, 4]))
            yield pages.AppropPUN_u, dict(pun_u=choice([1, 2, 3, 4]))
            yield pages.AppropPUN_d, dict(pun_d=choice([1, 2, 3, 4]))

        # if not not self.player.participant.vars['inactive']:
        yield pages.Payoff
        yield pages.Questionnaire, dict(age=24, gender=1, educ=4, student=1, left_right=6, experience=0,
                                        purpose="bla", unclear="bla2", didntwork="bla3")
        yield Submission(pages.End, check_html=False)