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

Island: Modify AgentConfiguration endpoint to use new pydantic model

This commit is contained in:
Shreya Malviya 2022-08-29 19:42:04 +05:30
parent c79b3c4497
commit 433e154cd1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/monkey_island/cc/resources/agent_configuration.py
View File

@ -20,13 +20,13 @@ class AgentConfiguration(AbstractResource):
# Used by the agent. Can't secure # Used by the agent. Can't secure
def get(self): def get(self):
configuration = self._agent_configuration_repository.get_configuration() configuration = self._agent_configuration_repository.get_configuration()
configuration_json = AgentConfigurationObject.to_json(configuration) configuration_dict = configuration.dict()
return make_response(configuration_json, 200) return make_response(configuration_dict, 200)
@jwt_required @jwt_required
def put(self): def put(self):
try: try:
configuration_object = AgentConfigurationObject.from_mapping(request.json) configuration_object = AgentConfigurationObject(**request.json)
self._agent_configuration_repository.store_configuration(configuration_object) self._agent_configuration_repository.store_configuration(configuration_object)
# API Spec: Should return 204 (NO CONTENT) # API Spec: Should return 204 (NO CONTENT)
return make_response({}, 200) return make_response({}, 200)