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

class PlayerBot(Bot):
    def play_round(self):
        if self.round_number==1:
            yield pages.P01_Welcome
            yield pages.P02_T1_Inst1
            yield pages.P04_T1_Inst2
            yield pages.P05_T1_Inst3
            yield pages.P07_T1_Control, dict(control1 = 0,
                                             control2 = 1,
                                             control3 = 0,
                                             control4 = 80,
                                             control5 = 20,
                                             control6 = 80)
            yield pages.P09_T1_Role
        if self.player.type == "seller":
            yield pages.Z17_T2seller, dict(T2_product_type = random.randint(0, 1), T2_price = random.choice(range(1, 49)))
            if self.player.id_in_group == 2:
                yield pages.Z19_InfoSeller1
            if self.player.id_in_group == 3:
                yield pages.Z20_InfoSeller2
        if self.player.type == "buyer":
            yield pages.Z21_T2buyer, dict(choice= random.choice(range(1, 3)))
        yield pages.Z24_T2feedback
        if self.round_number == 2:
            yield pages.Z25_T3intro
            yield pages.Z26_Q1, dict(age = random.choice(range(18, 30)), sex = random.randint(0, 1), exp = random.choice(range(0, 3)),
                                     subject = random.choice(["Economics", "Literature"]), nationality = "Something", city = "Something")
            yield pages.Z27_Q2, dict(markets1 = random.choice(range(1, 6)), markets2 = random.choice(range(1, 6)), markets3 = random.choice(range(1, 6)),
                                     ingroup_city = random.choice(range(1, 6)), ingroup_region = random.choice(range(1, 6)),
                                     events1 = random.choice(range(1, 6)), events2 = random.choice(range(1, 6)))
            if self.player.type == "buyer":
                yield pages.Z28_Q3, dict(verstanden = random.choice(range(1, 6)) , level_identity = random.choice(range(1, 6)),
                                     not_harming_member1 = random.choice(range(1, 6)), not_harming_member2 = random.choice(range(1, 6)),
                                     ownpayoff = random.choice(range(1, 6)), utilitarian = random.choice(range(1, 6)), identity = random.choice(range(1, 6)),
                                     quality_check = random.choice(range(1, 6)))
            if self.player.type == "seller":
                yield pages.Z28_Q3_seller, dict(verstanden = random.choice(range(1, 6)),
                                     not_harming_member1 = random.choice(range(1, 6)), not_harming_member2 = random.choice(range(1, 6)),
                                     ownpayoff = random.choice(range(1, 6)), utilitarian = random.choice(range(1, 6)),
                                     quality_check = random.choice(range(1, 6)))
            if self.player.type == "C":
                yield pages.Z28_Q3_C, dict(verstanden = random.choice(range(1, 6)), quality_check = random.choice(range(1, 6)))
            yield pages.Z29_Q4, dict(partei = random.choice(range(1, 7)) , pol_selbstein = random.choice(range(1, 10)))
            yield pages.Z30_ResultsT1,
            yield Submission(pages.Z31_Results, check_html = False)