from otree.api import Currency as c, currency_range, expect, Bot
from . import *
#from .init import Constants
import random
#from otree.api import Submission
#from otree.api import SubmissionMustFail


class PlayerBot(Bot):
    def play_round(player):
        
        yield Instructions
        
        yield Survey1, dict(gender=random.randint(1,3),
                            citizen=random.randint(1,3),
                            zipcode=random.randint(1,6))
        
        yield Survey2, dict(education=random.randint(1,6),
                            partyid=random.randint(1, 4),
                            partyid_imp=random.randint(1, 7)), 
                                                      
        yield Survey3, dict(racialid=random.randint(1, 6),
                            ethnicid=random.randint(1,6),
                            attention=1)
                            
        yield Survey4Trust, dict(govertrust=random.randint(1, 5),
                                 friendtrust=random.randint(1,5),
                                 generaltrust=random.randint(1,3))
                                 
        yield Survey5, dict(contact=random.randint(1,5),
                            discriminatio_per=random.randint(1,5),
                            discrimination_group=random.randint(1,5))
                            
        
        yield Submission(Survey6Ratings, dict(whiterating=random.randint(1, 100),
                         demorating=random.randint(1, 100),
                         blackrating=random.randint(1, 100),
                         asianrating=random.randint(1, 100),
                         immigrantrating=random.randint(1, 100), 
                         latinorating=random.randint(1, 100),
                         repubrating=random.randint(1, 100)),
                         check_html=False)
        
        yield Survey7, dict(politicalid=random.randint(1, 6),
                            americanid=random.randint(1, 6))