platypush/platypush/plugins/__init__.py
Fabio Manganiello a2898f63c3 - Unwrap the response in the decorator if we got a Response object from the decorated plugin
- No need for @action wrappers around omxplayer event handlers
2018-07-15 02:03:24 +02:00

49 lines
1.3 KiB
Python

import sys
import logging
import traceback
from platypush.config import Config
from platypush.message.response import Response
from platypush.utils import get_decorators
def action(f):
def _execute_action(*args, **kwargs):
output = None
errors = []
try:
output = f(*args, **kwargs)
if output and isinstance(output, Response):
errors = output.errors
output = output.output
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)
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__).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: