# Path: oscar/agent/custom_agent.py
# class CustomAgent(BaseAgent):
#     """A base agent to write custom scripted agents."""
# 
#     def __init__(self):
#         super().__init__()
#         self._shared = {}
# 
#     def add_shared(self, name, shared):
#         self._shared[name] = shared


import random
from oscar.agent.custom_agent import CustomAgent
from oscar.util.attack import *
from oscar.constants import *


class AttackManager(CustomAgent):

    def __init__(self, message="I hate you"):
        self._message = message
        super().__init__()

    def step(self, obs, locked_choice=None):
        play = {}

        enemy_point = get_random_enemy_location(obs)
        if enemy_point:

# Path: oscar/agent/custom_agent.py
# class CustomAgent(BaseAgent):
#     """A base agent to write custom scripted agents."""
# 
#     def __init__(self):
#         super().__init__()
#         self._shared = {}
# 
#     def add_shared(self, name, shared):
#         self._shared[name] = shared


# Path: oscar/agent/scripted/attack_manager.py
import random
from oscar.agent.custom_agent import CustomAgent
from oscar.util.attack import *
from oscar.constants import *



class AttackManager(CustomAgent):

    def __init__(self, message="I hate you"):
        self._message = message
        super().__init__()

    def step(self, obs, locked_choice=None):
        play = {}

        enemy_point = get_random_enemy_location(obs)
        if enemy_point:

            play['actions'] = [actions.FunctionCall(SELECT_ARMY, [NEW_SELECTION]),