forked from platypush/platypush
137 lines
3.8 KiB
Python
137 lines
3.8 KiB
Python
import functools
|
|
import importlib
|
|
import logging
|
|
import signal
|
|
import time
|
|
|
|
from platypush.config import Config
|
|
from platypush.message import Message
|
|
|
|
modules = {}
|
|
|
|
def get_or_load_plugin(plugin_name, reload=False):
|
|
global modules
|
|
|
|
if plugin_name in modules and not reload:
|
|
return modules[plugin_name]
|
|
|
|
try:
|
|
module = importlib.import_module('platypush.plugins.' + plugin_name)
|
|
except ModuleNotFoundError as e:
|
|
logging.warning('No such plugin: {}'.format(plugin_name))
|
|
raise RuntimeError(e)
|
|
|
|
# e.g. plugins.music.mpd main class: MusicMpdPlugin
|
|
cls_name = functools.reduce(
|
|
lambda a,b: a.title() + b.title(),
|
|
(plugin_name.title().split('.'))
|
|
) + 'Plugin'
|
|
|
|
plugin_conf = Config.get_plugins()[plugin_name] \
|
|
if plugin_name in Config.get_plugins() else {}
|
|
|
|
try:
|
|
plugin_class = getattr(module, cls_name)
|
|
plugin = plugin_class(**plugin_conf)
|
|
modules[plugin_name] = plugin
|
|
except AttributeError as e:
|
|
logging.warning('No such class in {}: {}'.format(
|
|
plugin_name, cls_name))
|
|
raise RuntimeError(e)
|
|
|
|
return plugin
|
|
|
|
|
|
def init_backends(bus=None, **kwargs):
|
|
""" Initialize the backend objects based on the configuration and returns
|
|
a name -> backend_instance map.
|
|
Params:
|
|
bus -- If specific (it usually should), the messages processed by the
|
|
backends will be posted on this bus.
|
|
|
|
kwargs -- Any additional key-value parameters required to initialize the backends
|
|
"""
|
|
|
|
backends = {}
|
|
|
|
for k in Config.get_backends().keys():
|
|
module = importlib.import_module('platypush.backend.' + k)
|
|
cfg = Config.get_backends()[k]
|
|
|
|
# e.g. backend.pushbullet main class: PushbulletBackend
|
|
cls_name = functools.reduce(
|
|
lambda a,b: a.title() + b.title(),
|
|
(module.__name__.title().split('.')[2:])
|
|
) + 'Backend'
|
|
|
|
try:
|
|
b = getattr(module, cls_name)(bus=bus, **cfg, **kwargs)
|
|
backends[k] = b
|
|
except AttributeError as e:
|
|
logging.warning('No such class in {}: {}'.format(
|
|
module.__name__, cls_name))
|
|
raise RuntimeError(e)
|
|
|
|
return backends
|
|
|
|
def get_module_and_name_from_action(action):
|
|
""" Input : action=music.mpd.play
|
|
Output : ('music.mpd', 'play') """
|
|
|
|
tokens = action.split('.')
|
|
module_name = str.join('.', tokens[:-1])
|
|
method_name = tokens[-1:][0]
|
|
return (module_name, method_name)
|
|
|
|
|
|
def get_message_class_by_type(msgtype):
|
|
""" Gets the class of a message type given as string """
|
|
|
|
try:
|
|
module = importlib.import_module('platypush.message.' + msgtype)
|
|
except ModuleNotFoundError as e:
|
|
logging.warning('Unsupported message type {}'.format(msgtype))
|
|
raise RuntimeError(e)
|
|
|
|
cls_name = msgtype[0].upper() + msgtype[1:]
|
|
|
|
try:
|
|
msgclass = getattr(module, cls_name)
|
|
except AttributeError as e:
|
|
logging.warning('No such class in {}: {}'.format(
|
|
module.__name__, cls_name))
|
|
raise RuntimeError(e)
|
|
|
|
return msgclass
|
|
|
|
|
|
def get_event_class_by_type(type):
|
|
""" Gets an event class by type name """
|
|
event_module = importlib.import_module('.'.join(type.split('.')[:-1]))
|
|
return getattr(event_module, type.split('.')[-1])
|
|
|
|
|
|
def set_timeout(seconds, on_timeout):
|
|
"""
|
|
Set a function to be called if timeout expires without being cleared.
|
|
It only works on the main thread.
|
|
|
|
Params:
|
|
seconds -- Timeout in seconds
|
|
on_timeout -- Function invoked on timeout unless clear_timeout is called before
|
|
"""
|
|
|
|
def _sighandler(signum, frame):
|
|
on_timeout()
|
|
|
|
signal.signal(signal.SIGALRM, _sighandler)
|
|
signal.alarm(seconds)
|
|
|
|
|
|
def clear_timeout():
|
|
""" Clear any previously set timeout """
|
|
signal.alarm(0)
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|
|
|