forked from platypush/platypush
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
import sys
|
|
import logging
|
|
import traceback
|
|
|
|
from functools import wraps
|
|
|
|
from platypush.config import Config
|
|
from platypush.message.response import Response
|
|
from platypush.utils import get_decorators
|
|
|
|
def action(f):
|
|
@wraps(f)
|
|
def _execute_action(*args, **kwargs):
|
|
output = None
|
|
errors = []
|
|
|
|
try:
|
|
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]
|
|
except Exception as e:
|
|
if isinstance(args[0], Plugin):
|
|
args[0].logger.exception(e)
|
|
errors.append(str(e) + '\n' + traceback.format_exc())
|
|
|
|
return Response(output=output, errors=errors)
|
|
|
|
# Propagate the docstring
|
|
_execute_action.__doc__ = f.__doc__
|
|
return _execute_action
|
|
|
|
|
|
class Plugin(object):
|
|
""" Base plugin class """
|
|
|
|
def __init__(self, **kwargs):
|
|
self.logger = logging.getLogger(self.__class__.__name__)
|
|
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)
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|
|
|