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

Island: Use UUID for AgentSignals endpoint parameter

This commit is contained in:
Ilija Lazoroski 2022-09-28 16:45:28 +02:00
parent 1a8306af1b
commit 10954e0a6e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/monkey_island/cc/resources/agent_signals/agent_signals.py
View File

@ -1,6 +1,7 @@
import logging import logging
from http import HTTPStatus from http import HTTPStatus
from common.types import AgentID
from monkey_island.cc.resources.AbstractResource import AbstractResource from monkey_island.cc.resources.AbstractResource import AbstractResource
from monkey_island.cc.services import AgentSignalsService from monkey_island.cc.services import AgentSignalsService
@ -8,7 +9,7 @@ logger = logging.getLogger(__name__)
class AgentSignals(AbstractResource): class AgentSignals(AbstractResource):
urls = ["/api/agent-signals/<string:agent_id>"] urls = ["/api/agent-signals/<uuid:agent_id>"]
def __init__( def __init__(
self, self,
@ -16,6 +17,6 @@ class AgentSignals(AbstractResource):
): ):
self._agent_signals_service = agent_signals_service self._agent_signals_service = agent_signals_service
def get(self, agent_id: str): def get(self, agent_id: AgentID):
agent_signals = self._agent_signals_service.get_signals(agent_id) agent_signals = self._agent_signals_service.get_signals(agent_id)
return agent_signals.dict(simplify=True), HTTPStatus.OK return agent_signals.dict(simplify=True), HTTPStatus.OK