from otree.api import Currency as c, currency_range, expect, Bot
from . import *
import random

class PlayerBot(Bot):
    def play_round(self): 
        
        if self.round_number == 1:
            yield Survey, dict(age=random.randint(1,5),
                           citizen=random.randint(1,2),
                           race=random.randint(1,5),
                           income=random.randint(1,6))
            yield Send, dict(contrib=random.randint(0,10))
                
            yield Results 
        else: 
            yield Send, dict(contrib=random.randint(0,30))
                
            yield Results