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

Agent: Fix incorrect return type on PluginRegistry.get_plugin()

This commit is contained in:
Mike Salvatore 2022-02-16 14:11:27 -05:00
parent 5953373125
commit 5b53984014
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/infection_monkey/puppet/plugin_registry.py
View File

@ -1,5 +1,4 @@
import logging import logging
from typing import Optional
from infection_monkey.i_puppet import PluginType, UnknownPluginError from infection_monkey.i_puppet import PluginType, UnknownPluginError
@ -28,7 +27,7 @@ class PluginRegistry:
logger.debug(f"Plugin '{plugin_name}' loaded") logger.debug(f"Plugin '{plugin_name}' loaded")
def get_plugin(self, plugin_name: str, plugin_type: PluginType) -> Optional[object]: def get_plugin(self, plugin_name: str, plugin_type: PluginType) -> object:
try: try:
plugin = self._registry[plugin_type][plugin_name] plugin = self._registry[plugin_type][plugin_name]
except KeyError: except KeyError: