import random
import actions as Act
class Bot:
max_tower = 30
def init(self, player_id):
self.player_id = player_id
def inject_data(self, data):
pass
def get_action(self, obs, bg = None, max_opt=3):
# if self.player_id == 0:
# obs['player_info'][1]['lives'] = 100
# print (obs['player_info'][1]['lives'])
action_seq = []
for i in range(max_opt):
action_seq.append(self._get_random_action())
return action_seq
def game_over(self, winner, obs):
pass
def _get_random_action(self):
d1 = random.choice(list(Act.Actions))
if d1 == Act.Actions.NOOP:
return Act.noop()
elif d1 == Act.Actions.BUY_TOWER:
type = random.randint(0, 1)
slot = random.randint(0, self.max_tower)
return Act.buy_tower(type, slot)
elif d1 == Act.Actions.BUY_MONSTER:
type = random.randint(0, 3)
return Act.buy_monster(type)