Island: Add docstrings to reset_agent_configuration event handler

This commit is contained in:
Ilija Lazoroski 2022-09-13 09:31:38 +02:00
parent ea47fc326a
commit 599c753cf7
1 changed files with 8 additions and 0 deletions

View File

@ -2,6 +2,11 @@ from monkey_island.cc.repository import IAgentConfigurationRepository, IFileRepo
class reset_agent_configuration: class reset_agent_configuration:
"""
Callable class that handles removal of PBA files and reset of agent configuration
on the Island.
"""
def __init__( def __init__(
self, self,
agent_configuration_repository: IAgentConfigurationRepository, agent_configuration_repository: IAgentConfigurationRepository,
@ -15,6 +20,9 @@ class reset_agent_configuration:
self._agent_configuration_repository.reset_to_default() self._agent_configuration_repository.reset_to_default()
def _remove_pba_files(self): def _remove_pba_files(self):
"""
Removes PBA files from the file repository
"""
agent_configuration = self._agent_configuration_repository.get_configuration() agent_configuration = self._agent_configuration_repository.get_configuration()
custom_pbas = agent_configuration.custom_pbas custom_pbas = agent_configuration.custom_pbas