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

class PlayerBot(Bot):

    cases = ['min', 'max']

    def play_round(self):
        yield (pages.Introduction)

        if self.case == 'min':
            yield (pages.Decide, {'units': 0})
            # if player produces 0, nothing is sold and they make 0
            assert self.player.payoff == c(0)

        if self.case == 'max':
            yield (pages.Decide, {'units': Constants.max_units_per_player})
            # if everyone produces max, price is driven to 0
            assert self.player.payoff == c(0)

        yield (pages.Results)