Mirrored_Repos/houdini
1
0
Fork 0
mirror of https://github.com/solero/houdini.git synced 2025-04-29 01:56:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support for plugin loading & unloading

Browse Source
This commit is contained in:
Ben 2019-04-11 01:04:46 +01:00
parent dddeeeed99
commit 39398bd6b1
1 changed files with 40 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
Houdini/HoudiniFactory.py
View File

@ -31,8 +31,10 @@ except ImportError:
uvloop = None uvloop = None
import Houdini.Handlers import Houdini.Handlers
from Houdini.Handlers import listeners_from_module from Houdini.Handlers import listeners_from_module, remove_handlers_by_module
from Houdini.Events.HandlerFileEvent import HandlerFileEventHandler from Houdini.Events.HandlerFileEvent import HandlerFileEventHandler
from Houdini.Events.PluginFileEvent import PluginFileEventHandler
import Houdini.Plugins as Plugins
class HoudiniFactory: class HoudiniFactory:
@ -143,13 +145,46 @@ class HoudiniFactory:
handlers_path = './Houdini{}Handlers'.format(os.path.sep) handlers_path = './Houdini{}Handlers'.format(os.path.sep)
plugins_path = './Houdini{}Plugins'.format(os.path.sep) plugins_path = './Houdini{}Plugins'.format(os.path.sep)
self.configure_obvservers([handlers_path, HandlerFileEventHandler]) self.configure_obvservers([handlers_path, HandlerFileEventHandler],
[plugins_path, PluginFileEventHandler])
self.logger.info('Listening on {}:{}'.format(self.server_config['Address'], self.server_config['Port'])) self.logger.info('Listening on {}:{}'.format(self.server_config['Address'], self.server_config['Port']))
await self.load_plugins()
async with self.server: async with self.server:
await self.server.serve_forever() await self.server.serve_forever()
async def load_plugins(self):
for plugin_package in self.get_package_modules(Plugins):
await self.load_plugin(plugin_package)
async def load_plugin(self, plugin):
plugin_module, plugin_class = plugin
if plugin_class not in self.server_config['Plugins']:
return
plugin_object = getattr(plugin_module, plugin_class)(self)
if isinstance(plugin_object, Plugins.IPlugin):
self.plugins[plugin_class] = plugin_object
listeners_from_module(self.xt_listeners, self.xml_listeners, plugin_object)
await plugin_object.ready()
else:
self.logger.warn('{0} plugin object doesn\'t provide the plugin interface'.format(plugin_class))
def unload_plugin(self, plugin):
plugin_module, plugin_class = plugin
if plugin_class in self.plugins:
plugin_module_path = plugin_module.__file__
del self.plugins[plugin_class]
remove_handlers_by_module(self.xt_listeners, self.xml_listeners, plugin_module_path)
async def client_connected(self, reader, writer): async def client_connected(self, reader, writer):
client_object = self.client_class(self, reader, writer) client_object = self.client_class(self, reader, writer)
await client_object.run() await client_object.run()
@ -161,19 +196,19 @@ class HoudiniFactory:
module = importlib.import_module(handler_module) module = importlib.import_module(handler_module)
listeners_from_module(self.xt_listeners, self.xml_listeners, module) listeners_from_module(self.xt_listeners, self.xml_listeners, module)
self.logger.info("Handler modules loaded") self.logger.info('Handler modules loaded')
def get_package_modules(self, package): def get_package_modules(self, package):
package_modules = [] package_modules = []
for importer, module_name, is_package in pkgutil.iter_modules(package.__path__): for importer, module_name, is_package in pkgutil.iter_modules(package.__path__):
full_module_name = "{0}.{1}".format(package.__name__, module_name) full_module_name = '{0}.{1}'.format(package.__name__, module_name)
if is_package: if is_package:
subpackage_object = importlib.import_module(full_module_name, package=package.__path__) subpackage_object = importlib.import_module(full_module_name, package=package.__path__)
subpackage_object_directory = dir(subpackage_object) subpackage_object_directory = dir(subpackage_object)
if "Plugin" in subpackage_object_directory: if Plugins.IPlugin.__name__ in subpackage_object_directory:
package_modules.append((subpackage_object, module_name)) package_modules.append((subpackage_object, module_name))
continue continue