from otree.api import Currency as cu, currency_range

from . import *
from otree.api import Bot


class PlayerBot(Bot):

    def play_round(self):
        yield (SurveyIntro)
        if self.player.participant.vars['id_in_group']==2:
            yield (Employee_1, {'m_1':1,'e_q1':1, 'e_q2':2, 'e_q3':3,'e_q4':1, 'e_q5':2, 'e_q6':3,'e_q7':1})
            yield (Employee_2, {'m_2':1,'e_q8': 1, 'e_q9': 2, 'e_q10': 3, 'e_q11': 1, 'e_q12': 2, 'e_q13': 3})
        else:
            yield (Supervisor_1, {'m_1':1,'s_q1': 1, 's_q2': 2, 's_q3': 3, 's_q4': 1, 's_q5': 2, 's_q6': 3})
            yield (Supervisor_2, {'m_2':1,'s_q7': 1,'s_q8': 1, 's_q9': 2, 's_q10': 3, 's_q11': 1})
        yield (Tom, {'t_q1':1, 't_q2':2, 't_q3':3,'t_q4':1, 't_q5':2, 't_q6':3,'t_q7':1})
        yield (Demographics, {
            'age': 24,
            'gender': 1,
            'study_field':'accounting',
               'work_experience':3,
            'experience_experiments':4}
            )