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.player.participant.vars['treatment'] == 0:
            yield pages.Questionnaire, dict(instructions=True,
                                            fair_play=1,
                                            strategy="Being fair")
        else:
            if self.player.participant.vars['type'] == "A":
                yield pages.Questionnaire, dict(instructions=True,
                                                fair_play=1,
                                                strategy="Being fair",
                                                worried_A=1,
                                                main_belief_A=1,
                                                effect_strategy_A_1=True,
                                                effect_strategy_belief_A_1=True)
            else:
                yield pages.Questionnaire, dict(instructions=True,
                                                fair_play=1,
                                                strategy="Being fair",
                                                worried_belief_B=1,
                                                temptation_B=1,
                                                main_motive_B=1,
                                                further_motive_B_1=1,
                                                main_belief_B=1,
                                                effect_strategy_B_1=True,
                                                effect_strategy_belief_B_1=True)
            yield pages.Results