revil-O/platypush
1
0
Fork 0
forked from platypush/platypush
Code Pull requests Activity

Caching modules and plugins

Browse source
This commit is contained in:
Fabio Manganiello 2017-11-03 02:42:56 +01:00
parent 2de28153f0
commit 13023b660e
1 changed files with 24 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
notifier.py
View file

@ -34,6 +34,9 @@ DEVICE_ID = config['device_id'] \
DEBUG = config['debug'] if 'debug' in config else False DEBUG = config['debug'] if 'debug' in config else False
modules = {}
plugins = {}
def on_open(ws): def on_open(ws):
logging.info('Connection opened') logging.info('Connection opened')
@ -48,9 +51,13 @@ def on_error(ws, error):
def _exec_func(body): def _exec_func(body):
module = None module_name = 'plugins.{}'.format(body['plugin'])
if module_name in modules:
module = modules[module_name]
else:
try: try:
module = importlib.import_module('plugins.{}'.format(body['plugin'])) module = importlib.import_module(module_name)
modules[module_name] = module
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
logging.warn('No such plugin: {}'.format(body['plugin'])) logging.warn('No such plugin: {}'.format(body['plugin']))
return return
@ -58,14 +65,18 @@ def _exec_func(body):
logging.info('Received push addressed to me: {}'.format(body)) logging.info('Received push addressed to me: {}'.format(body))
args = body['args'] if 'args' in body else {} args = body['args'] if 'args' in body else {}
cls = getattr( cls_name = functools.reduce(
module, functools.reduce(
lambda a,b: a.title() + b.title(), lambda a,b: a.title() + b.title(),
(body['plugin'].title().split('.')) (body['plugin'].title().split('.'))
) + 'Plugin' ) + 'Plugin'
)
if cls_name in plugins:
instance = plugins[cls_name]
else:
cls = getattr(module, cls_name)
instance = cls() instance = cls()
plugins[cls_name] = cls
out, err = instance.run(args) out, err = instance.run(args)
logging.info('Command output: {}'.format(out)) logging.info('Command output: {}'.format(out))