p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Change agent parameter to agent_id in get_signals()

This commit is contained in:
Mike Salvatore 2022-09-22 12:59:51 -04:00
parent f7997a6a50
commit 8e45a71a15
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/monkey_island/cc/services/agent_signals_service.py
View File

@ -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())