diff --git a/monkey/monkey_island/cc/island_event_handlers/__init__.py b/monkey/monkey_island/cc/island_event_handlers/__init__.py index 7a20d4c44..0c977976e 100644 --- a/monkey/monkey_island/cc/island_event_handlers/__init__.py +++ b/monkey/monkey_island/cc/island_event_handlers/__init__.py @@ -1 +1,2 @@ from .reset_agent_configuration import reset_agent_configuration +from .reset_machine_repository import reset_machine_repository diff --git a/monkey/monkey_island/cc/island_event_handlers/reset_machine_repository.py b/monkey/monkey_island/cc/island_event_handlers/reset_machine_repository.py new file mode 100644 index 000000000..b62060f47 --- /dev/null +++ b/monkey/monkey_island/cc/island_event_handlers/reset_machine_repository.py @@ -0,0 +1,14 @@ +from monkey_island.cc.repository import IMachineRepository, initialize_machine_repository + + +class reset_machine_repository: + """ + Callable class that handles reset and reinitialization of IMachineRepository + """ + + def __init__(self, machine_repository: IMachineRepository): + self._machine_repository = machine_repository + + def __call__(self): + self._machine_repository.reset() + initialize_machine_repository(self._machine_repository)