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


class PlayerBot(Bot):

    # TODO: add is_displayed logic
    def play_round(self):
        yield (pages.TaskCInstructions)
        yield Submission(pages.TaskC, {'creativity': 'test'}, check_html=False)
        yield (pages.TaskIInstructions)
        yield Submission(pages.TaskI, {'iat_tables': 'test'}, check_html=False)
        yield (pages.TaskTrDecision, {'trolley_decision': 0})
        yield (pages.TaskADecision, {'ambiguity_decision': 'A'})
        bigfive_fields = {}
        for i in Constants.bigfive_items:
            bigfive_fields.update({i: 1})
        yield (pages.TaskBF, bigfive_fields)
        yield (pages.TaskGInstructions, {'grit_decision': 'easy'})
        yield (pages.TaskG, {'grit_num_correct': 3})
        yield (pages.TaskGResults)
        yield (pages.RsQuestions, {'risky_smoke': 1, 'risky_drink': 1})
        yield (pages.TaskScDecision, {'self_confidence': 'test'})
        yield (pages.TaskWDecision, {"wason_card_d": False, "wason_card_k": False, "wason_card_3": False, "wason_card_7": False})
        yield (pages.TaskGngInstructions)
        yield Submission(pages.TaskGng, {'gonogo_trials': 'test'}, check_html=False)
        empathy_fields = {}
        for n in range(len(Constants.empathy_labels)):
            empathy_fields.update({f'empathy_{n}': 1})
        yield (pages.TaskEm, empathy_fields)
        curiosity_fields = {}
        for n in range(len(Constants.curiosity_labels)):
            curiosity_fields.update({f'curiosity_{n}': 1})
        yield (pages.TaskCr, curiosity_fields)
        if self.participant.vars.get('do_rating_creativity'):
            yield (pages.RatingCInstructions)
            yield (pages.RatingC, {'creativity_rating': 'L'})
            if self.player.creativity_rating_with_incentive:
                yield (pages.RatingCEstimationInstructions)
                yield (pages.RatingCEstimation, {'creativity_rating_estimate': 'L'})
        yield (pages.Tech, {'technical_issues': False, 'distractions': True})
        yield (pages.Email, {'email': ''})
        yield (pages.Comments, {'comments': 'test'})