Throw an assert error (no reload/retry mechanism) instead of a runtime error when a method is called outside of the registered plugin actions
This commit is contained in:
parent
2541cab0f7
commit
ddbe76646b
1 changed files with 3 additions and 6 deletions
|
@ -45,15 +45,12 @@ class Plugin(EventGenerator):
|
||||||
self.logger.setLevel(getattr(logging, kwargs['logging'].upper()))
|
self.logger.setLevel(getattr(logging, kwargs['logging'].upper()))
|
||||||
|
|
||||||
self.registered_actions = set(
|
self.registered_actions = set(
|
||||||
get_decorators(self.__class__, climb_class_hierarchy=True)
|
get_decorators(self.__class__, climb_class_hierarchy=True).get('action', [])
|
||||||
.get('action', [])
|
|
||||||
)
|
)
|
||||||
|
|
||||||
def run(self, method, *args, **kwargs):
|
def run(self, method, *args, **kwargs):
|
||||||
if method not in self.registered_actions:
|
assert method in self.registered_actions, '{} is not a registered action on {}'.\
|
||||||
raise RuntimeError('{} is not a registered action on {}'.format(
|
format(method, self.__class__.__name__)
|
||||||
method, self.__class__.__name__))
|
|
||||||
|
|
||||||
return getattr(self, method)(*args, **kwargs)
|
return getattr(self, method)(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue