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

Agent: Use Iterable instead of List for type hint in run_plugins()

This commit is contained in:
Mike Salvatore 2022-01-26 08:13:35 -05:00
parent 8cf54e7673
commit 92636da4b2
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/infection_monkey/master/automated_master.py
View File

@ -1,7 +1,7 @@
import logging
import threading
import time
from typing import Any, Callable, Dict, List, Tuple
from typing import Any, Callable, Dict, Iterable, List, Tuple
from infection_monkey.i_control_channel import IControlChannel, IslandCommunicationError
from infection_monkey.i_master import IMaster
@ -191,7 +191,9 @@ class AutomatedMaster(IMaster):
self._puppet.run_payload(name, options, self._stop)
def _run_plugins(self, plugins: List[Any], plugin_type: str, callback: Callable[[Any], None]):
def _run_plugins(
self, plugins: Iterable[Any], plugin_type: str, callback: Callable[[Any], None]
):
logger.info(f"Running {plugin_type}s")
logger.debug(f"Found {len(plugins)} {plugin_type}(s) to run")