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

Island: Add AgentID type for agent_id to AgentLogs endpoint

This commit is contained in:
Ilija Lazoroski 2022-09-28 16:21:39 +02:00
parent fa13ca8df8
commit b705e33af3
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/monkey_island/cc/resources/agent_logs.py
View File

@ -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/<string:agent_id>"]
urls = ["/api/agent-logs/<uuid:agent_id>"]
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