diff options
Diffstat (limited to 'pytest_helpers_namespace')
-rw-r--r-- | pytest_helpers_namespace/plugin.py | 18 | ||||
-rw-r--r-- | pytest_helpers_namespace/version.py | 4 |
2 files changed, 14 insertions, 8 deletions
diff --git a/pytest_helpers_namespace/plugin.py b/pytest_helpers_namespace/plugin.py index 729036b..95ab3e5 100644 --- a/pytest_helpers_namespace/plugin.py +++ b/pytest_helpers_namespace/plugin.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- ''' :codeauthor: :email:`Pedro Algarvio (pedro@algarvio.me)` - :copyright: © 2016 by the SaltStack Team, see AUTHORS for more details. + :copyright: © 2016-2017 by the SaltStack Team, see AUTHORS for more details. :license: Apache 2.0, see LICENSE for more details. @@ -12,7 +12,8 @@ ''' # Import python libs -from functools import wraps +from functools import partial, wraps + class FuncWrapper(object): @@ -53,17 +54,22 @@ class HelpersRegistry(object): def __init__(self): self._registry = {} - def register(self, func): + def register(self, func, name=None): ''' Register's a new function as a helper ''' - if func.__name__ in self._registry: + if isinstance(func, str): + return partial(self.register, name=func) + + if name is None: + name = func.__name__ + if name in self._registry: raise RuntimeError( 'A helper function is already registered under the name: {0}'.format( - func.__name__ + name ) ) - self._registry[func.__name__] = wraps(func)(FuncWrapper(func)) + self._registry[name] = wraps(func)(FuncWrapper(func)) return func def __getattribute__(self, name): diff --git a/pytest_helpers_namespace/version.py b/pytest_helpers_namespace/version.py index 6e9d74b..12bc370 100644 --- a/pytest_helpers_namespace/version.py +++ b/pytest_helpers_namespace/version.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- ''' :codeauthor: :email:`Pedro Algarvio (pedro@algarvio.me)` - :copyright: © 2016 by the SaltStack Team, see AUTHORS for more details. + :copyright: © 2016-2017 by the SaltStack Team, see AUTHORS for more details. :license: Apache 2.0, see LICENSE for more details. @@ -16,5 +16,5 @@ from __future__ import absolute_import -__version_info__ = (2016, 7, 10) +__version_info__ = (2017, 11, 11) __version__ = '{0}.{1}.{2}'.format(*__version_info__) |