From 35025ead183e7bd11110c1cecb6df22ced58c577 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 20 Sep 2022 14:33:44 -0400 Subject: [PATCH] Island: Add reset_machine_repository --- .../cc/island_event_handlers/__init__.py | 1 + .../reset_machine_repository.py | 14 ++++++++++++++ 2 files changed, 15 insertions(+) create mode 100644 monkey/monkey_island/cc/island_event_handlers/reset_machine_repository.py 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)