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

class PlayerBot(Bot):    
    def play_round(self):
        response = random.choice([1, 1, 1, 0])
        offer = random.choice([1, 1, 1, 0])
        if self.player.role == C.OFFER_ROLE:
            yield Submission(Offer, dict(offer=offer), check_html=False)
        else:
            yield Submission(Response, dict(response=response), check_html=False)
        tokens_for_offered = random.randint(0, 10)
        tokens_for_notoffered = 10 - tokens_for_offered
        if self.player.role == C.RESPONDER_ROLE:
            yield Submission(Beliefs, dict(tokens_for_offered=tokens_for_offered,
                                       tokens_for_notoffered=tokens_for_notoffered),
                         check_html=False)