From 8e45a71a155ee3ed8e37e5029be36b68173c7c57 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Thu, 22 Sep 2022 12:59:51 -0400 Subject: [PATCH] Island: Change agent parameter to agent_id in get_signals() --- monkey/monkey_island/cc/services/agent_signals_service.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/monkey/monkey_island/cc/services/agent_signals_service.py b/monkey/monkey_island/cc/services/agent_signals_service.py index 47c677271..4a3c4ef74 100644 --- a/monkey/monkey_island/cc/services/agent_signals_service.py +++ b/monkey/monkey_island/cc/services/agent_signals_service.py @@ -1,18 +1,19 @@ from datetime import datetime -from monkey_island.cc.models import Agent, AgentSignals +from common.types import AgentID from monkey_island.cc.repository import ISimulationRepository +from monkey_island.cc.models import AgentSignals class AgentSignalsService: def __init__(self, simulation_repository: ISimulationRepository): self._simulation_repository = simulation_repository - def get_signals(self, agent: Agent) -> AgentSignals: + def get_signals(self, agent_id: AgentID) -> AgentSignals: """ Gets the signals sent to a particular agent - :param agent: The agent whose signals need to be retrieved + :param agent_id: The ID of the agent whose signals need to be retrieved :return: Signals sent to the relevant agent """ return AgentSignals(timestamp=datetime.now())