From b705e33af390897595fe6011cd55615115edf24e Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Wed, 28 Sep 2022 16:21:39 +0200 Subject: [PATCH] Island: Add AgentID type for agent_id to AgentLogs endpoint --- monkey/monkey_island/cc/resources/agent_logs.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/monkey/monkey_island/cc/resources/agent_logs.py b/monkey/monkey_island/cc/resources/agent_logs.py index b9c7b1957..e6bb287a9 100644 --- a/monkey/monkey_island/cc/resources/agent_logs.py +++ b/monkey/monkey_island/cc/resources/agent_logs.py @@ -2,25 +2,27 @@ from http import HTTPStatus from flask import request +from common.types import AgentID from monkey_island.cc.repository import IAgentLogRepository from monkey_island.cc.resources.AbstractResource import AbstractResource from monkey_island.cc.resources.request_authentication import jwt_required class AgentLogs(AbstractResource): - urls = ["/api/agent-logs/"] + urls = ["/api/agent-logs/"] def __init__(self, agent_log_repository: IAgentLogRepository): self._agent_log_repository = agent_log_repository @jwt_required - def get(self, agent_id: str): + def get(self, agent_id: AgentID): log_contents = self._agent_log_repository.get_agent_log(agent_id) return log_contents, HTTPStatus.OK - def put(self, agent_id: str): - log_contents = request.json["log_contents"] - self._agent_log_repository.upsert_agent_log(agent_id, log_contents) + def put(self, agent_id: AgentID): + log_contents = request.json + + self._agent_log_repository.upsert_agent_log(agent_id, agent_data) return {}, HTTPStatus.NO_CONTENT