mirror of
https://github.com/solero/houdini.git
synced 2024-11-10 05:08:19 +00:00
80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
import sys
|
|
import importlib
|
|
from watchdog.events import FileSystemEventHandler
|
|
|
|
from Houdini.Handlers import listeners_from_module
|
|
from Houdini.Events import evaluate_handler_file_event, remove_handlers_by_module
|
|
|
|
|
|
class HandlerFileEventHandler(FileSystemEventHandler):
|
|
|
|
def __init__(self, server):
|
|
self.logger = server.logger
|
|
self.server = server
|
|
|
|
def on_created(self, event):
|
|
handler_module_details = evaluate_handler_file_event(event)
|
|
|
|
if not handler_module_details:
|
|
return
|
|
|
|
handler_module_path, handler_module = handler_module_details
|
|
|
|
if "__init__.py" in handler_module_path:
|
|
return
|
|
|
|
self.logger.debug("New handler module detected %s", handler_module)
|
|
|
|
try:
|
|
module = importlib.import_module(handler_module)
|
|
listeners_from_module(self.server.xt_listeners, self.server.xml_listeners, module)
|
|
except Exception as import_error:
|
|
self.logger.error("%s detected in %s, not importing.", import_error.__class__.__name__, handler_module)
|
|
|
|
def on_deleted(self, event):
|
|
handler_module_details = evaluate_handler_file_event(event)
|
|
|
|
if not handler_module_details:
|
|
return
|
|
|
|
handler_module_path, handler_module = handler_module_details
|
|
|
|
if handler_module not in sys.modules:
|
|
return
|
|
|
|
self.logger.debug("Deleting listeners registered by %s..", handler_module)
|
|
|
|
remove_handlers_by_module(self.server.xt_listeners, self.server.xml_listeners, handler_module_path)
|
|
|
|
def on_modified(self, event):
|
|
handler_module_details = evaluate_handler_file_event(event)
|
|
|
|
if not handler_module_details:
|
|
return
|
|
|
|
handler_module_path, handler_module = handler_module_details
|
|
|
|
if handler_module not in sys.modules:
|
|
return False
|
|
|
|
self.logger.info("Reloading %s", handler_module)
|
|
|
|
xt_listeners, xml_listeners = remove_handlers_by_module(self.server.xt_listeners, self.server.xml_listeners,
|
|
handler_module_path)
|
|
|
|
handler_module_object = sys.modules[handler_module]
|
|
|
|
try:
|
|
module = importlib.reload(handler_module_object)
|
|
listeners_from_module(self.server.xt_listeners, self.server.xml_listeners, module)
|
|
|
|
self.logger.info("Successfully reloaded %s!", handler_module)
|
|
except Exception as rebuild_error:
|
|
self.logger.error("%s detected in %s, not reloading.", rebuild_error.__class__.__name__, handler_module)
|
|
self.logger.info("Restoring handler references...")
|
|
|
|
self.server.xt_listeners = xt_listeners
|
|
self.server.xml_listeners = xml_listeners
|
|
|
|
self.logger.info("Handler references restored. Phew!")
|