platypush/platypush/plugins/__init__.py

59 lines
1.6 KiB
Python
Raw Normal View History

import sys
2017-12-11 03:53:26 +01:00
import logging
import traceback
2018-11-01 19:42:40 +01:00
from functools import wraps
2017-12-18 01:10:51 +01:00
from platypush.config import Config
from platypush.message.response import Response
from platypush.utils import get_decorators
def action(f):
2018-11-01 19:42:40 +01:00
@wraps(f)
def _execute_action(*args, **kwargs):
output = None
errors = []
output = f(*args, **kwargs)
if output and isinstance(output, Response):
errors = output.errors \
if isinstance(output.errors, list) else [output.errors]
output = output.output
elif isinstance(output, tuple) and len(output) == 2:
errors = output[1] \
if isinstance(output[1], list) else [output[1]]
if len(errors) == 1 and errors[0] is None: errors = []
output = output[0]
return Response(output=output, errors=errors)
# Propagate the docstring
_execute_action.__doc__ = f.__doc__
return _execute_action
2017-10-31 09:20:35 +01:00
class Plugin(object):
2017-12-18 01:10:51 +01:00
""" Base plugin class """
2017-11-03 15:06:29 +01:00
2017-12-18 01:10:51 +01:00
def __init__(self, **kwargs):
self.logger = logging.getLogger(self.__class__.__name__)
2018-06-06 20:09:18 +02:00
if 'logging' in kwargs:
self.logger.setLevel(getattr(logging, kwargs['logging'].upper()))
self.registered_actions = set(
get_decorators(self.__class__, climb_class_hierarchy=True)
.get('action', [])
)
def run(self, method, *args, **kwargs):
if method not in self.registered_actions:
raise RuntimeError('{} is not a registered action on {}'.format(
method, self.__class__.__name__))
return getattr(self, method)(*args, **kwargs)
2017-10-31 09:20:35 +01:00
# vim:sw=4:ts=4:et: