2008-08-05 02:39:50 -05:00
|
|
|
# Authors:
|
|
|
|
# Jason Gerard DeRose <jderose@redhat.com>
|
|
|
|
#
|
|
|
|
# Copyright (C) 2008 Red Hat
|
|
|
|
# see file 'COPYING' for use and warranty information
|
|
|
|
#
|
|
|
|
# This program is free software; you can redistribute it and/or
|
|
|
|
# modify it under the terms of the GNU General Public License as
|
|
|
|
# published by the Free Software Foundation; version 2 only
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
# along with this program; if not, write to the Free Software
|
|
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
|
|
|
|
"""
|
2008-09-23 19:12:35 -05:00
|
|
|
Base classes for all front-end plugins.
|
2008-08-05 02:39:50 -05:00
|
|
|
"""
|
|
|
|
|
|
|
|
import re
|
2008-08-10 19:21:12 -05:00
|
|
|
import inspect
|
2008-08-05 02:39:50 -05:00
|
|
|
import plugable
|
2008-09-02 12:41:55 -05:00
|
|
|
from plugable import lock, check_name
|
2008-08-06 22:38:49 -05:00
|
|
|
import errors
|
2008-09-03 13:32:49 -05:00
|
|
|
from errors import check_type, check_isinstance, raise_TypeError
|
2008-08-28 13:31:06 -05:00
|
|
|
import ipa_types
|
2008-08-06 22:38:49 -05:00
|
|
|
|
|
|
|
|
2008-08-06 23:51:21 -05:00
|
|
|
RULE_FLAG = 'validation_rule'
|
|
|
|
|
|
|
|
def rule(obj):
|
2008-08-08 12:11:29 -05:00
|
|
|
assert not hasattr(obj, RULE_FLAG)
|
|
|
|
setattr(obj, RULE_FLAG, True)
|
|
|
|
return obj
|
2008-08-06 23:51:21 -05:00
|
|
|
|
|
|
|
def is_rule(obj):
|
2008-08-08 12:11:29 -05:00
|
|
|
return callable(obj) and getattr(obj, RULE_FLAG, False) is True
|
2008-08-06 23:51:21 -05:00
|
|
|
|
|
|
|
|
2008-08-22 15:07:17 -05:00
|
|
|
class DefaultFrom(plugable.ReadOnly):
|
2008-08-25 20:07:24 -05:00
|
|
|
"""
|
2008-09-26 17:52:15 -05:00
|
|
|
Derive a default value from other supplied values.
|
2008-08-25 20:07:24 -05:00
|
|
|
|
2008-09-26 17:52:15 -05:00
|
|
|
For example, say you wanted to create a default for the user's login from
|
|
|
|
the user's first and last names. It could be implemented like this:
|
2008-08-25 20:07:24 -05:00
|
|
|
|
2008-09-26 17:52:15 -05:00
|
|
|
>>> login = DefaultFrom(lambda first, last: first[0] + last)
|
|
|
|
>>> login(first='John', last='Doe')
|
|
|
|
'JDoe'
|
|
|
|
|
|
|
|
If you do not explicitly provide keys when you create a DefaultFrom
|
|
|
|
instance, the keys are implicitly derived from your callback by
|
|
|
|
inspecting ``callback.func_code.co_varnames``. The keys are available
|
|
|
|
through the ``DefaultFrom.keys`` instance attribute, like this:
|
|
|
|
|
|
|
|
>>> login.keys
|
|
|
|
('first', 'last')
|
|
|
|
|
|
|
|
The callback is available through the ``DefaultFrom.callback`` instance
|
|
|
|
attribute, like this:
|
|
|
|
|
2008-10-17 21:50:34 -05:00
|
|
|
>>> login.callback # doctest:+ELLIPSIS
|
|
|
|
<function <lambda> at 0x...>
|
2008-09-26 17:52:15 -05:00
|
|
|
>>> login.callback.func_code.co_varnames # The keys
|
|
|
|
('first', 'last')
|
|
|
|
|
|
|
|
The keys can be explicitly provided as optional positional arguments after
|
|
|
|
the callback. For example, this is equivalent to the ``login`` instance
|
|
|
|
above:
|
|
|
|
|
|
|
|
>>> login2 = DefaultFrom(lambda a, b: a[0] + b, 'first', 'last')
|
|
|
|
>>> login2.keys
|
|
|
|
('first', 'last')
|
|
|
|
>>> login2.callback.func_code.co_varnames # Not the keys
|
|
|
|
('a', 'b')
|
|
|
|
>>> login2(first='John', last='Doe')
|
|
|
|
'JDoe'
|
|
|
|
|
|
|
|
If any keys are missing when calling your DefaultFrom instance, your
|
|
|
|
callback is not called and None is returned. For example:
|
|
|
|
|
|
|
|
>>> login(first='John', lastname='Doe') is None
|
2008-08-25 20:07:24 -05:00
|
|
|
True
|
2008-09-26 17:52:15 -05:00
|
|
|
>>> login() is None
|
2008-08-25 20:07:24 -05:00
|
|
|
True
|
2008-09-26 17:52:15 -05:00
|
|
|
|
|
|
|
Any additional keys are simply ignored, like this:
|
|
|
|
|
|
|
|
>>> login(last='Doe', first='John', middle='Whatever')
|
|
|
|
'JDoe'
|
|
|
|
|
|
|
|
As above, because `DefaultFrom.__call__` takes only pure keyword
|
|
|
|
arguments, they can be supplied in any order.
|
|
|
|
|
|
|
|
Of course, the callback need not be a lambda expression. This third
|
|
|
|
example is equivalent to both the ``login`` and ``login2`` instances
|
|
|
|
above:
|
|
|
|
|
|
|
|
>>> def get_login(first, last):
|
|
|
|
... return first[0] + last
|
|
|
|
...
|
|
|
|
>>> login3 = DefaultFrom(get_login)
|
|
|
|
>>> login3.keys
|
|
|
|
('first', 'last')
|
|
|
|
>>> login3.callback.func_code.co_varnames
|
|
|
|
('first', 'last')
|
|
|
|
>>> login3(first='John', last='Doe')
|
|
|
|
'JDoe'
|
2008-08-25 20:07:24 -05:00
|
|
|
"""
|
2008-09-26 17:52:15 -05:00
|
|
|
|
2008-08-22 15:07:17 -05:00
|
|
|
def __init__(self, callback, *keys):
|
2008-08-25 20:07:24 -05:00
|
|
|
"""
|
2008-09-26 17:52:15 -05:00
|
|
|
:param callback: The callable to call when all keys are present.
|
|
|
|
:param keys: Optional keys used for source values.
|
2008-08-25 20:07:24 -05:00
|
|
|
"""
|
2008-09-24 16:46:37 -05:00
|
|
|
if not callable(callback):
|
|
|
|
raise TypeError('callback must be callable; got %r' % callback)
|
2008-08-22 15:07:17 -05:00
|
|
|
self.callback = callback
|
2008-09-24 16:46:37 -05:00
|
|
|
if len(keys) == 0:
|
2008-10-20 17:45:32 -05:00
|
|
|
fc = callback.func_code
|
|
|
|
self.keys = fc.co_varnames[:fc.co_argcount]
|
2008-09-24 16:46:37 -05:00
|
|
|
else:
|
|
|
|
self.keys = keys
|
|
|
|
for key in self.keys:
|
|
|
|
if type(key) is not str:
|
|
|
|
raise_TypeError(key, str, 'keys')
|
2008-08-22 15:07:17 -05:00
|
|
|
lock(self)
|
|
|
|
|
|
|
|
def __call__(self, **kw):
|
2008-08-26 11:52:46 -05:00
|
|
|
"""
|
|
|
|
If all keys are present, calls the callback; otherwise returns None.
|
|
|
|
|
|
|
|
:param kw: The keyword arguments.
|
|
|
|
"""
|
2008-08-22 15:07:17 -05:00
|
|
|
vals = tuple(kw.get(k, None) for k in self.keys)
|
|
|
|
if None in vals:
|
2008-09-24 16:46:37 -05:00
|
|
|
return
|
2008-08-22 15:07:17 -05:00
|
|
|
try:
|
2008-08-26 11:52:46 -05:00
|
|
|
return self.callback(*vals)
|
2008-09-24 16:46:37 -05:00
|
|
|
except StandardError:
|
|
|
|
pass
|
2008-08-22 15:07:17 -05:00
|
|
|
|
|
|
|
|
2008-09-24 12:55:29 -05:00
|
|
|
def parse_param_spec(spec):
|
|
|
|
"""
|
2008-09-24 13:44:43 -05:00
|
|
|
Parse a param spec into to (name, kw).
|
2008-09-24 12:55:29 -05:00
|
|
|
|
|
|
|
The ``spec`` string determines the param name, whether the param is
|
|
|
|
required, and whether the param is multivalue according the following
|
|
|
|
syntax:
|
|
|
|
|
2008-09-26 18:41:51 -05:00
|
|
|
====== ===== ======== ==========
|
|
|
|
Spec Name Required Multivalue
|
|
|
|
====== ===== ======== ==========
|
|
|
|
'var' 'var' True False
|
|
|
|
'var?' 'var' False False
|
|
|
|
'var*' 'var' False True
|
|
|
|
'var+' 'var' True True
|
|
|
|
====== ===== ======== ==========
|
|
|
|
|
|
|
|
For example,
|
|
|
|
|
|
|
|
>>> parse_param_spec('login')
|
|
|
|
('login', {'required': True, 'multivalue': False})
|
|
|
|
>>> parse_param_spec('gecos?')
|
|
|
|
('gecos', {'required': False, 'multivalue': False})
|
|
|
|
>>> parse_param_spec('telephone_numbers*')
|
|
|
|
('telephone_numbers', {'required': False, 'multivalue': True})
|
|
|
|
>>> parse_param_spec('group+')
|
|
|
|
('group', {'required': True, 'multivalue': True})
|
2008-09-24 12:55:29 -05:00
|
|
|
|
|
|
|
:param spec: A spec string.
|
|
|
|
"""
|
|
|
|
if type(spec) is not str:
|
|
|
|
raise_TypeError(spec, str, 'spec')
|
|
|
|
if len(spec) < 2:
|
|
|
|
raise ValueError(
|
|
|
|
'param spec must be at least 2 characters; got %r' % spec
|
|
|
|
)
|
|
|
|
_map = {
|
|
|
|
'?': dict(required=False, multivalue=False),
|
|
|
|
'*': dict(required=False, multivalue=True),
|
|
|
|
'+': dict(required=True, multivalue=True),
|
|
|
|
}
|
|
|
|
end = spec[-1]
|
|
|
|
if end in _map:
|
|
|
|
return (spec[:-1], _map[end])
|
|
|
|
return (spec, dict(required=True, multivalue=False))
|
|
|
|
|
|
|
|
|
2008-09-21 17:02:33 -05:00
|
|
|
class Param(plugable.ReadOnly):
|
2008-09-24 14:45:46 -05:00
|
|
|
"""
|
|
|
|
A parameter accepted by a `Command`.
|
2008-09-26 19:31:59 -05:00
|
|
|
|
|
|
|
============ ================= ==================
|
|
|
|
Keyword Type Default
|
|
|
|
============ ================= ==================
|
2008-10-13 18:24:23 -05:00
|
|
|
cli_name str defaults to name
|
2008-09-26 19:31:59 -05:00
|
|
|
type ipa_type.Type ipa_type.Unicode()
|
2008-10-18 01:16:22 -05:00
|
|
|
doc str ""
|
2008-09-26 19:31:59 -05:00
|
|
|
required bool True
|
|
|
|
multivalue bool False
|
|
|
|
primary_key bool False
|
|
|
|
normalize callable None
|
|
|
|
default same as type.type None
|
|
|
|
default_from callable None
|
2008-10-17 20:34:26 -05:00
|
|
|
flags frozenset frozenset()
|
2008-09-26 19:31:59 -05:00
|
|
|
============ ================= ==================
|
2008-09-24 14:45:46 -05:00
|
|
|
"""
|
2008-09-24 02:56:31 -05:00
|
|
|
__nones = (None, '', tuple(), [])
|
2008-09-24 17:05:01 -05:00
|
|
|
__defaults = dict(
|
2008-10-13 18:24:23 -05:00
|
|
|
cli_name=None,
|
2008-09-26 20:30:39 -05:00
|
|
|
type=ipa_types.Unicode(),
|
2008-09-24 13:02:00 -05:00
|
|
|
doc='',
|
|
|
|
required=True,
|
|
|
|
multivalue=False,
|
2008-09-26 19:31:59 -05:00
|
|
|
primary_key=False,
|
|
|
|
normalize=None,
|
2008-09-24 13:02:00 -05:00
|
|
|
default=None,
|
|
|
|
default_from=None,
|
2008-10-17 20:34:26 -05:00
|
|
|
flags=frozenset(),
|
2008-09-24 13:02:00 -05:00
|
|
|
rules=tuple(),
|
|
|
|
)
|
2008-09-24 02:56:31 -05:00
|
|
|
|
2008-09-26 20:30:39 -05:00
|
|
|
def __init__(self, name, **override):
|
2008-09-24 16:57:34 -05:00
|
|
|
if not ('required' in override or 'multivalue' in override):
|
2008-09-24 13:27:14 -05:00
|
|
|
(name, kw_from_spec) = parse_param_spec(name)
|
2008-09-24 16:57:34 -05:00
|
|
|
override.update(kw_from_spec)
|
2008-09-24 17:05:01 -05:00
|
|
|
kw = dict(self.__defaults)
|
2008-10-13 18:24:23 -05:00
|
|
|
kw['cli_name'] = name
|
2008-09-24 16:57:34 -05:00
|
|
|
if not set(kw).issuperset(override):
|
|
|
|
extra = sorted(set(override) - set(kw))
|
2008-09-24 13:27:14 -05:00
|
|
|
raise TypeError(
|
2008-09-24 13:33:25 -05:00
|
|
|
'Param.__init__() takes no such kwargs: %s' % ', '.join(extra)
|
2008-09-24 13:27:14 -05:00
|
|
|
)
|
2008-09-24 16:57:34 -05:00
|
|
|
kw.update(override)
|
|
|
|
self.__kw = kw
|
2008-09-02 12:41:55 -05:00
|
|
|
self.name = check_name(name)
|
2008-10-13 18:24:23 -05:00
|
|
|
self.cli_name = check_name(kw.get('cli_name', name))
|
2008-09-26 20:30:39 -05:00
|
|
|
self.type = self.__check_isinstance(ipa_types.Type, 'type')
|
2008-09-24 13:27:14 -05:00
|
|
|
self.doc = self.__check_type(str, 'doc')
|
|
|
|
self.required = self.__check_type(bool, 'required')
|
|
|
|
self.multivalue = self.__check_type(bool, 'multivalue')
|
2008-09-24 16:57:34 -05:00
|
|
|
self.default = kw['default']
|
2008-09-24 17:05:01 -05:00
|
|
|
df = kw['default_from']
|
|
|
|
if callable(df) and not isinstance(df, DefaultFrom):
|
|
|
|
df = DefaultFrom(df)
|
|
|
|
self.default_from = check_type(df, DefaultFrom, 'default_from',
|
2008-09-24 13:27:14 -05:00
|
|
|
allow_none=True
|
|
|
|
)
|
2008-10-17 20:34:26 -05:00
|
|
|
self.flags = frozenset(kw['flags'])
|
2008-09-24 16:57:34 -05:00
|
|
|
self.__normalize = kw['normalize']
|
2008-09-24 14:45:46 -05:00
|
|
|
self.rules = self.__check_type(tuple, 'rules')
|
2008-09-26 20:30:39 -05:00
|
|
|
self.all_rules = (self.type.validate,) + self.rules
|
2008-09-24 16:29:15 -05:00
|
|
|
self.primary_key = self.__check_type(bool, 'primary_key')
|
2008-08-28 13:31:06 -05:00
|
|
|
lock(self)
|
|
|
|
|
2008-09-24 14:45:46 -05:00
|
|
|
def __clone__(self, **override):
|
|
|
|
"""
|
|
|
|
Return a new `Param` instance similar to this one.
|
|
|
|
"""
|
|
|
|
kw = dict(self.__kw)
|
|
|
|
kw.update(override)
|
2008-09-26 20:30:39 -05:00
|
|
|
return self.__class__(self.name, **kw)
|
2008-09-24 14:45:46 -05:00
|
|
|
|
2008-09-24 13:27:14 -05:00
|
|
|
def __check_type(self, type_, name, allow_none=False):
|
|
|
|
value = self.__kw[name]
|
|
|
|
return check_type(value, type_, name, allow_none)
|
|
|
|
|
|
|
|
def __check_isinstance(self, type_, name, allow_none=False):
|
|
|
|
value = self.__kw[name]
|
|
|
|
return check_isinstance(value, type_, name, allow_none)
|
|
|
|
|
2008-09-24 01:36:48 -05:00
|
|
|
def __dispatch(self, value, scalar):
|
2008-09-24 02:56:31 -05:00
|
|
|
"""
|
|
|
|
Helper method used by `normalize` and `convert`.
|
|
|
|
"""
|
|
|
|
if value in self.__nones:
|
2008-09-24 02:05:43 -05:00
|
|
|
return
|
2008-09-24 01:25:12 -05:00
|
|
|
if self.multivalue:
|
|
|
|
if type(value) in (tuple, list):
|
2008-09-24 01:35:19 -05:00
|
|
|
return tuple(
|
|
|
|
scalar(v, i) for (i, v) in enumerate(value)
|
|
|
|
)
|
|
|
|
return (scalar(value, 0),) # tuple
|
2008-09-24 01:25:12 -05:00
|
|
|
return scalar(value)
|
|
|
|
|
2008-09-24 01:35:19 -05:00
|
|
|
def __normalize_scalar(self, value, index=None):
|
2008-09-24 02:56:31 -05:00
|
|
|
"""
|
|
|
|
Normalize a scalar value.
|
|
|
|
|
|
|
|
This method is called once with each value in multivalue.
|
|
|
|
"""
|
2008-09-24 01:11:46 -05:00
|
|
|
if not isinstance(value, basestring):
|
|
|
|
return value
|
|
|
|
try:
|
|
|
|
return self.__normalize(value)
|
|
|
|
except StandardError:
|
|
|
|
return value
|
|
|
|
|
|
|
|
def normalize(self, value):
|
|
|
|
"""
|
|
|
|
Normalize ``value`` using normalize callback.
|
|
|
|
|
2008-10-18 01:16:22 -05:00
|
|
|
For example:
|
|
|
|
|
|
|
|
>>> param = Param('telephone',
|
|
|
|
... normalize=lambda value: value.replace('.', '-')
|
|
|
|
... )
|
|
|
|
>>> param.normalize('800.123.4567')
|
|
|
|
'800-123-4567'
|
|
|
|
|
2008-09-24 01:11:46 -05:00
|
|
|
If this `Param` instance does not have a normalize callback,
|
|
|
|
``value`` is returned unchanged.
|
|
|
|
|
|
|
|
If this `Param` instance has a normalize callback and ``value`` is
|
|
|
|
a basestring, the normalize callback is called and its return value
|
|
|
|
is returned.
|
|
|
|
|
|
|
|
If ``value`` is not a basestring, or if an exception is caught
|
|
|
|
when calling the normalize callback, ``value`` is returned unchanged.
|
|
|
|
|
|
|
|
:param value: A proposed value for this parameter.
|
|
|
|
"""
|
|
|
|
if self.__normalize is None:
|
|
|
|
return value
|
2008-09-24 01:36:48 -05:00
|
|
|
return self.__dispatch(value, self.__normalize_scalar)
|
2008-09-24 01:11:46 -05:00
|
|
|
|
2008-09-03 18:21:26 -05:00
|
|
|
def __convert_scalar(self, value, index=None):
|
2008-09-24 02:56:31 -05:00
|
|
|
"""
|
|
|
|
Convert a scalar value.
|
|
|
|
|
|
|
|
This method is called once with each value in multivalue.
|
|
|
|
"""
|
|
|
|
if value in self.__nones:
|
2008-09-24 02:05:43 -05:00
|
|
|
return
|
2008-09-03 13:32:49 -05:00
|
|
|
converted = self.type(value)
|
|
|
|
if converted is None:
|
|
|
|
raise errors.ConversionError(
|
2008-09-03 18:21:26 -05:00
|
|
|
self.name, value, self.type, index=index
|
2008-09-03 13:32:49 -05:00
|
|
|
)
|
|
|
|
return converted
|
|
|
|
|
2008-08-28 22:17:26 -05:00
|
|
|
def convert(self, value):
|
2008-09-24 01:48:27 -05:00
|
|
|
"""
|
2008-09-24 02:05:43 -05:00
|
|
|
Convert/coerce ``value`` to Python type for this `Param`.
|
|
|
|
|
2008-10-18 01:16:22 -05:00
|
|
|
For example:
|
|
|
|
|
|
|
|
>>> param = Param('an_int', type=ipa_types.Int())
|
|
|
|
>>> param.convert(7.2)
|
|
|
|
7
|
|
|
|
>>> param.convert(" 7 ")
|
|
|
|
7
|
|
|
|
|
2008-09-24 02:56:31 -05:00
|
|
|
If ``value`` can not be converted, ConversionError is raised, which
|
|
|
|
is as subclass of ValidationError.
|
2008-09-24 02:05:43 -05:00
|
|
|
|
|
|
|
If ``value`` is None, conversion is not attempted and None is
|
|
|
|
returned.
|
2008-08-28 22:17:26 -05:00
|
|
|
|
2008-09-24 01:48:27 -05:00
|
|
|
:param value: A proposed value for this parameter.
|
|
|
|
"""
|
|
|
|
return self.__dispatch(value, self.__convert_scalar)
|
2008-08-28 22:17:26 -05:00
|
|
|
|
2008-09-03 14:38:39 -05:00
|
|
|
def __validate_scalar(self, value, index=None):
|
2008-09-24 02:56:31 -05:00
|
|
|
"""
|
|
|
|
Validate a scalar value.
|
|
|
|
|
|
|
|
This method is called once with each value in multivalue.
|
|
|
|
"""
|
2008-09-03 14:38:39 -05:00
|
|
|
if type(value) is not self.type.type:
|
|
|
|
raise_TypeError(value, self.type.type, 'value')
|
2008-08-28 13:31:06 -05:00
|
|
|
for rule in self.rules:
|
2008-08-28 15:30:08 -05:00
|
|
|
error = rule(value)
|
|
|
|
if error is not None:
|
2008-09-03 20:01:40 -05:00
|
|
|
raise errors.RuleError(
|
|
|
|
self.name, value, error, rule, index=index
|
|
|
|
)
|
2008-08-28 13:31:06 -05:00
|
|
|
|
|
|
|
def validate(self, value):
|
2008-09-24 02:56:31 -05:00
|
|
|
"""
|
|
|
|
Check validity of a value.
|
|
|
|
|
|
|
|
Each validation rule is called in turn and if any returns and error,
|
|
|
|
RuleError is raised, which is a subclass of ValidationError.
|
|
|
|
|
|
|
|
:param value: A proposed value for this parameter.
|
|
|
|
"""
|
|
|
|
if value is None:
|
|
|
|
if self.required:
|
|
|
|
raise errors.RequirementError(self.name)
|
|
|
|
return
|
2008-08-28 13:31:06 -05:00
|
|
|
if self.multivalue:
|
|
|
|
if type(value) is not tuple:
|
2008-09-03 20:01:40 -05:00
|
|
|
raise_TypeError(value, tuple, 'value')
|
|
|
|
for (i, v) in enumerate(value):
|
|
|
|
self.__validate_scalar(v, i)
|
2008-08-28 13:31:06 -05:00
|
|
|
else:
|
2008-08-28 22:17:26 -05:00
|
|
|
self.__validate_scalar(value)
|
2008-08-28 13:31:06 -05:00
|
|
|
|
2008-09-02 14:05:10 -05:00
|
|
|
def get_default(self, **kw):
|
2008-09-24 02:56:31 -05:00
|
|
|
"""
|
|
|
|
Return a default value for this parameter.
|
|
|
|
|
|
|
|
If this `Param` instance does not have a default_from() callback, this
|
|
|
|
method always returns the static Param.default instance attribute.
|
|
|
|
|
|
|
|
On the other hand, if this `Param` instance has a default_from()
|
|
|
|
callback, the callback is called and its return value is returned
|
|
|
|
(assuming that value is not None).
|
|
|
|
|
|
|
|
If the default_from() callback returns None, or if an exception is
|
|
|
|
caught when calling the default_from() callback, the static
|
|
|
|
Param.default instance attribute is returned.
|
|
|
|
|
|
|
|
:param kw: Optional keyword arguments to pass to default_from().
|
|
|
|
"""
|
2008-09-02 14:05:10 -05:00
|
|
|
if self.default_from is not None:
|
|
|
|
default = self.default_from(**kw)
|
|
|
|
if default is not None:
|
2008-09-04 03:33:41 -05:00
|
|
|
try:
|
|
|
|
return self.convert(self.normalize(default))
|
|
|
|
except errors.ValidationError:
|
|
|
|
return None
|
2008-09-03 21:41:31 -05:00
|
|
|
return self.default
|
2008-09-02 14:05:10 -05:00
|
|
|
|
2008-09-02 14:29:00 -05:00
|
|
|
def get_values(self):
|
2008-10-18 01:16:22 -05:00
|
|
|
"""
|
|
|
|
Return a tuple of possible values.
|
|
|
|
|
|
|
|
For enumerable types, a tuple containing the possible values is
|
|
|
|
returned. For all other types, an empty tuple is returned.
|
|
|
|
"""
|
2008-09-02 14:29:00 -05:00
|
|
|
if self.type.name in ('Enum', 'CallbackEnum'):
|
|
|
|
return self.type.values
|
|
|
|
return tuple()
|
|
|
|
|
2008-09-04 02:18:26 -05:00
|
|
|
def __call__(self, value, **kw):
|
2008-09-24 02:56:31 -05:00
|
|
|
if value in self.__nones:
|
2008-09-04 03:16:12 -05:00
|
|
|
value = self.get_default(**kw)
|
|
|
|
else:
|
2008-09-04 03:33:41 -05:00
|
|
|
value = self.convert(self.normalize(value))
|
2008-09-24 02:56:31 -05:00
|
|
|
self.validate(value)
|
|
|
|
return value
|
2008-09-04 02:18:26 -05:00
|
|
|
|
2008-09-08 16:51:05 -05:00
|
|
|
def __repr__(self):
|
2008-09-09 19:21:40 -05:00
|
|
|
return '%s(%r, %s())' % (
|
2008-09-08 16:51:05 -05:00
|
|
|
self.__class__.__name__,
|
|
|
|
self.name,
|
|
|
|
self.type.name,
|
|
|
|
)
|
|
|
|
|
2008-08-28 13:31:06 -05:00
|
|
|
|
2008-09-21 17:43:50 -05:00
|
|
|
def create_param(spec):
|
2008-09-09 20:03:59 -05:00
|
|
|
"""
|
2008-09-24 13:44:43 -05:00
|
|
|
Create a `Param` instance from a param spec.
|
2008-09-21 17:18:33 -05:00
|
|
|
|
2008-09-21 17:43:50 -05:00
|
|
|
If ``spec`` is a `Param` instance, ``spec`` is returned unchanged.
|
2008-09-21 17:18:33 -05:00
|
|
|
|
2008-09-21 17:43:50 -05:00
|
|
|
If ``spec`` is an str instance, then ``spec`` is parsed and an
|
|
|
|
appropriate `Param` instance is created and returned.
|
|
|
|
|
2008-09-24 13:44:43 -05:00
|
|
|
See `parse_param_spec` for the definition of the spec syntax.
|
2008-09-21 17:43:50 -05:00
|
|
|
|
|
|
|
:param spec: A spec string or a `Param` instance.
|
2008-09-09 20:03:59 -05:00
|
|
|
"""
|
2008-09-21 17:43:50 -05:00
|
|
|
if type(spec) is Param:
|
|
|
|
return spec
|
|
|
|
if type(spec) is not str:
|
|
|
|
raise TypeError(
|
|
|
|
'create_param() takes %r or %r; got %r' % (str, Param, spec)
|
|
|
|
)
|
2008-09-24 13:44:43 -05:00
|
|
|
return Param(spec)
|
2008-09-09 20:03:59 -05:00
|
|
|
|
|
|
|
|
2008-08-15 14:49:04 -05:00
|
|
|
class Command(plugable.Plugin):
|
2008-10-08 19:01:22 -05:00
|
|
|
"""
|
|
|
|
A public IPA atomic operation.
|
|
|
|
|
|
|
|
All plugins that subclass from `Command` will be automatically available
|
|
|
|
as a CLI command and as an XML-RPC method.
|
|
|
|
|
|
|
|
Plugins that subclass from Command are registered in the ``api.Command``
|
|
|
|
namespace. For example:
|
|
|
|
|
|
|
|
>>> api = plugable.API(Command)
|
|
|
|
>>> class my_command(Command):
|
|
|
|
... pass
|
|
|
|
...
|
|
|
|
>>> api.register(my_command)
|
|
|
|
>>> api.finalize()
|
|
|
|
>>> list(api.Command)
|
|
|
|
['my_command']
|
2008-10-17 21:50:34 -05:00
|
|
|
>>> api.Command.my_command # doctest:+ELLIPSIS
|
|
|
|
PluginProxy(Command, ...my_command())
|
2008-10-08 19:01:22 -05:00
|
|
|
"""
|
|
|
|
|
2008-08-08 12:11:29 -05:00
|
|
|
__public__ = frozenset((
|
2008-08-26 14:13:55 -05:00
|
|
|
'get_default',
|
2008-09-03 21:30:40 -05:00
|
|
|
'convert',
|
2008-09-02 18:40:44 -05:00
|
|
|
'normalize',
|
2008-08-11 12:37:33 -05:00
|
|
|
'validate',
|
2008-08-11 14:11:26 -05:00
|
|
|
'execute',
|
2008-08-08 16:40:03 -05:00
|
|
|
'__call__',
|
2008-09-09 20:54:48 -05:00
|
|
|
'args',
|
2008-09-10 10:14:26 -05:00
|
|
|
'options',
|
2008-09-21 11:59:12 -05:00
|
|
|
'params',
|
2008-09-18 19:00:54 -05:00
|
|
|
'args_to_kw',
|
|
|
|
'kw_to_args',
|
2008-10-14 02:45:30 -05:00
|
|
|
'output_for_cli',
|
2008-08-08 12:11:29 -05:00
|
|
|
))
|
2008-09-10 09:46:20 -05:00
|
|
|
takes_options = tuple()
|
2008-09-08 20:41:15 -05:00
|
|
|
takes_args = tuple()
|
2008-09-21 13:50:00 -05:00
|
|
|
args = None
|
|
|
|
options = None
|
|
|
|
params = None
|
2008-10-17 22:05:03 -05:00
|
|
|
output_for_cli = None
|
2008-08-08 12:11:29 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def __call__(self, *args, **kw):
|
|
|
|
"""
|
|
|
|
Perform validation and then execute the command.
|
2008-09-09 18:46:16 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
If not in a server context, the call will be forwarded over
|
|
|
|
XML-RPC and the executed an the nearest IPA server.
|
|
|
|
"""
|
|
|
|
if len(args) > 0:
|
|
|
|
arg_kw = self.args_to_kw(*args)
|
|
|
|
assert set(arg_kw).intersection(kw) == set()
|
|
|
|
kw.update(arg_kw)
|
|
|
|
kw = self.normalize(**kw)
|
|
|
|
kw = self.convert(**kw)
|
|
|
|
kw.update(self.get_default(**kw))
|
|
|
|
self.validate(**kw)
|
|
|
|
args = tuple(kw.pop(name) for name in self.args)
|
|
|
|
return self.run(*args, **kw)
|
2008-09-09 16:18:44 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def args_to_kw(self, *values):
|
|
|
|
"""
|
|
|
|
Map positional into keyword arguments.
|
|
|
|
"""
|
|
|
|
if self.max_args is not None and len(values) > self.max_args:
|
|
|
|
if self.max_args == 0:
|
|
|
|
raise errors.ArgumentError(self, 'takes no arguments')
|
|
|
|
if self.max_args == 1:
|
|
|
|
raise errors.ArgumentError(self, 'takes at most 1 argument')
|
|
|
|
raise errors.ArgumentError(self,
|
|
|
|
'takes at most %d arguments' % len(self.args)
|
|
|
|
)
|
|
|
|
return dict(self.__args_to_kw_iter(values))
|
2008-08-08 12:11:29 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def __args_to_kw_iter(self, values):
|
|
|
|
"""
|
|
|
|
Generator used by `Command.args_to_kw` method.
|
|
|
|
"""
|
2008-09-09 18:46:16 -05:00
|
|
|
multivalue = False
|
2008-10-08 19:01:22 -05:00
|
|
|
for (i, arg) in enumerate(self.args()):
|
|
|
|
assert not multivalue
|
|
|
|
if len(values) > i:
|
|
|
|
if arg.multivalue:
|
|
|
|
multivalue = True
|
|
|
|
yield (arg.name, values[i:])
|
|
|
|
else:
|
|
|
|
yield (arg.name, values[i])
|
|
|
|
else:
|
|
|
|
break
|
2008-09-09 18:46:16 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def kw_to_args(self, **kw):
|
|
|
|
"""
|
|
|
|
Map keyword into positional arguments.
|
|
|
|
"""
|
|
|
|
return tuple(kw.get(name, None) for name in self.args)
|
2008-09-10 10:14:26 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def normalize(self, **kw):
|
|
|
|
"""
|
|
|
|
Return a dictionary of normalized values.
|
|
|
|
|
|
|
|
For example:
|
|
|
|
|
|
|
|
>>> class my_command(Command):
|
|
|
|
... takes_options = (
|
|
|
|
... Param('first', normalize=lambda value: value.lower()),
|
|
|
|
... Param('last'),
|
|
|
|
... )
|
|
|
|
...
|
|
|
|
>>> c = my_command()
|
|
|
|
>>> c.finalize()
|
|
|
|
>>> c.normalize(first='JOHN', last='DOE')
|
|
|
|
{'last': 'DOE', 'first': 'john'}
|
|
|
|
"""
|
2008-09-24 00:46:49 -05:00
|
|
|
return dict(
|
2008-10-08 19:01:22 -05:00
|
|
|
(k, self.params[k].normalize(v)) for (k, v) in kw.iteritems()
|
2008-09-24 00:46:49 -05:00
|
|
|
)
|
2008-09-03 21:30:40 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def convert(self, **kw):
|
|
|
|
"""
|
|
|
|
Return a dictionary of values converted to correct type.
|
|
|
|
|
|
|
|
>>> from ipalib import ipa_types
|
|
|
|
>>> class my_command(Command):
|
|
|
|
... takes_args = (
|
|
|
|
... Param('one', type=ipa_types.Int()),
|
|
|
|
... 'two',
|
|
|
|
... )
|
|
|
|
...
|
|
|
|
>>> c = my_command()
|
|
|
|
>>> c.finalize()
|
|
|
|
>>> c.convert(one=1, two=2)
|
|
|
|
{'two': u'2', 'one': 1}
|
|
|
|
"""
|
2008-09-24 00:49:30 -05:00
|
|
|
return dict(
|
2008-10-08 19:01:22 -05:00
|
|
|
(k, self.params[k].convert(v)) for (k, v) in kw.iteritems()
|
2008-09-24 00:49:30 -05:00
|
|
|
)
|
2008-08-08 12:11:29 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def get_default(self, **kw):
|
|
|
|
"""
|
|
|
|
Return a dictionary of defaults for all missing required values.
|
|
|
|
|
|
|
|
For example:
|
|
|
|
|
|
|
|
>>> class my_command(Command):
|
|
|
|
... takes_args = [Param('color', default='Red')]
|
|
|
|
...
|
|
|
|
>>> c = my_command()
|
|
|
|
>>> c.finalize()
|
|
|
|
>>> c.get_default()
|
|
|
|
{'color': 'Red'}
|
|
|
|
>>> c.get_default(color='Yellow')
|
|
|
|
{}
|
|
|
|
"""
|
|
|
|
return dict(self.__get_default_iter(kw))
|
|
|
|
|
2008-09-02 18:40:44 -05:00
|
|
|
def __get_default_iter(self, kw):
|
2008-10-08 19:01:22 -05:00
|
|
|
"""
|
|
|
|
Generator method used by `Command.get_default`.
|
|
|
|
"""
|
2008-09-10 19:04:49 -05:00
|
|
|
for param in self.params():
|
2008-11-12 02:47:37 -06:00
|
|
|
if kw.get(param.name, None) is None:
|
|
|
|
if param.required:
|
|
|
|
yield (param.name, param.get_default(**kw))
|
|
|
|
else:
|
|
|
|
yield (param.name, None)
|
2008-08-11 12:57:07 -05:00
|
|
|
|
2008-08-11 12:37:33 -05:00
|
|
|
def validate(self, **kw):
|
2008-10-08 19:01:22 -05:00
|
|
|
"""
|
|
|
|
Validate all values.
|
|
|
|
|
|
|
|
If any value fails the validation, `ipalib.errors.ValidationError`
|
|
|
|
(or a subclass thereof) will be raised.
|
|
|
|
"""
|
2008-09-10 19:04:49 -05:00
|
|
|
for param in self.params():
|
|
|
|
value = kw.get(param.name, None)
|
2008-09-03 21:02:06 -05:00
|
|
|
if value is not None:
|
2008-09-10 19:04:49 -05:00
|
|
|
param.validate(value)
|
|
|
|
elif param.required:
|
|
|
|
raise errors.RequirementError(param.name)
|
2008-08-11 14:11:26 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def run(self, *args, **kw):
|
|
|
|
"""
|
|
|
|
Dispatch to `Command.execute` or `Command.forward`.
|
|
|
|
|
|
|
|
If running in a server context, `Command.execute` is called and the
|
|
|
|
actually work this command performs is executed locally.
|
|
|
|
|
|
|
|
If running in a non-server context, `Command.forward` is called,
|
|
|
|
which forwards this call over XML-RPC to the exact same command
|
|
|
|
on the nearest IPA server and the actual work this command
|
|
|
|
performs is executed remotely.
|
|
|
|
"""
|
2008-10-27 02:35:40 -05:00
|
|
|
if self.api.env.in_server:
|
2008-10-08 19:01:22 -05:00
|
|
|
target = self.execute
|
|
|
|
else:
|
|
|
|
target = self.forward
|
|
|
|
object.__setattr__(self, 'run', target)
|
|
|
|
return target(*args, **kw)
|
|
|
|
|
2008-09-21 20:28:57 -05:00
|
|
|
def execute(self, *args, **kw):
|
2008-10-08 19:01:22 -05:00
|
|
|
"""
|
|
|
|
Perform the actual work this command does.
|
|
|
|
|
|
|
|
This method should be implemented only against functionality
|
|
|
|
in self.api.Backend. For example, a hypothetical
|
|
|
|
user_add.execute() might be implemented like this:
|
|
|
|
|
|
|
|
>>> class user_add(Command):
|
|
|
|
... def execute(self, **kw):
|
|
|
|
... return self.api.Backend.ldap.add(**kw)
|
|
|
|
...
|
|
|
|
"""
|
2008-10-08 19:18:13 -05:00
|
|
|
raise NotImplementedError('%s.execute()' % self.name)
|
2008-08-11 12:37:33 -05:00
|
|
|
|
2008-09-23 21:52:19 -05:00
|
|
|
def forward(self, *args, **kw):
|
2008-10-02 20:42:06 -05:00
|
|
|
"""
|
2008-10-08 19:01:22 -05:00
|
|
|
Forward call over XML-RPC to this same command on server.
|
2008-10-02 20:42:06 -05:00
|
|
|
"""
|
2008-11-12 02:47:37 -06:00
|
|
|
return self.Backend.xmlrpc.forward_call(self.name, *args, **kw)
|
2008-10-02 18:02:24 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def finalize(self):
|
|
|
|
"""
|
|
|
|
Finalize plugin initialization.
|
2008-09-23 21:52:19 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
This method creates the ``args``, ``options``, and ``params``
|
|
|
|
namespaces. This is not done in `Command.__init__` because
|
|
|
|
subclasses (like `crud.Add`) might need to access other plugins
|
|
|
|
loaded in self.api to determine what their custom `Command.get_args`
|
|
|
|
and `Command.get_options` methods should yield.
|
|
|
|
"""
|
|
|
|
self.args = plugable.NameSpace(self.__create_args(), sort=False)
|
|
|
|
if len(self.args) == 0 or not self.args[-1].multivalue:
|
|
|
|
self.max_args = len(self.args)
|
2008-09-23 21:52:19 -05:00
|
|
|
else:
|
2008-10-08 19:01:22 -05:00
|
|
|
self.max_args = None
|
|
|
|
self.options = plugable.NameSpace(
|
|
|
|
(create_param(spec) for spec in self.get_options()),
|
|
|
|
sort=False
|
|
|
|
)
|
2008-10-13 22:53:03 -05:00
|
|
|
def get_key(p):
|
|
|
|
if p.required:
|
|
|
|
if p.default_from is None:
|
|
|
|
return 0
|
|
|
|
return 1
|
|
|
|
return 2
|
2008-10-08 19:01:22 -05:00
|
|
|
self.params = plugable.NameSpace(
|
2008-10-13 22:53:03 -05:00
|
|
|
sorted(tuple(self.args()) + tuple(self.options()), key=get_key),
|
|
|
|
sort=False
|
2008-10-08 19:01:22 -05:00
|
|
|
)
|
|
|
|
super(Command, self).finalize()
|
2008-08-05 16:10:49 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def get_args(self):
|
|
|
|
"""
|
|
|
|
Return iterable with arguments for Command.args namespace.
|
2008-09-14 18:17:36 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
Subclasses can override this to customize how the arguments
|
|
|
|
are determined. For an example of why this can be useful,
|
|
|
|
see `ipalib.crud.Mod`.
|
|
|
|
"""
|
|
|
|
return self.takes_args
|
2008-09-18 14:39:23 -05:00
|
|
|
|
2008-10-08 19:01:22 -05:00
|
|
|
def get_options(self):
|
|
|
|
"""
|
|
|
|
Return iterable with options for Command.options namespace.
|
|
|
|
|
|
|
|
Subclasses can override this to customize how the options
|
|
|
|
are determined. For an example of why this can be useful,
|
|
|
|
see `ipalib.crud.Mod`.
|
|
|
|
"""
|
|
|
|
return self.takes_options
|
|
|
|
|
|
|
|
def __create_args(self):
|
|
|
|
"""
|
|
|
|
Generator used to create args namespace.
|
|
|
|
"""
|
|
|
|
optional = False
|
|
|
|
multivalue = False
|
|
|
|
for arg in self.get_args():
|
|
|
|
arg = create_param(arg)
|
|
|
|
if optional and arg.required:
|
|
|
|
raise ValueError(
|
|
|
|
'%s: required argument after optional' % arg.name
|
|
|
|
)
|
|
|
|
if multivalue:
|
|
|
|
raise ValueError(
|
|
|
|
'%s: only final argument can be multivalue' % arg.name
|
|
|
|
)
|
|
|
|
if not arg.required:
|
|
|
|
optional = True
|
|
|
|
if arg.multivalue:
|
|
|
|
multivalue = True
|
|
|
|
yield arg
|
2008-09-04 02:18:26 -05:00
|
|
|
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-08-22 16:50:53 -05:00
|
|
|
class Object(plugable.Plugin):
|
2008-08-08 12:11:29 -05:00
|
|
|
__public__ = frozenset((
|
2008-09-25 21:43:11 -05:00
|
|
|
'backend',
|
2008-09-24 17:19:43 -05:00
|
|
|
'methods',
|
2008-09-24 18:19:34 -05:00
|
|
|
'properties',
|
2008-09-24 21:13:16 -05:00
|
|
|
'params',
|
2008-09-24 20:44:53 -05:00
|
|
|
'primary_key',
|
2008-09-24 22:27:40 -05:00
|
|
|
'params_minus_pk',
|
2008-10-14 00:26:24 -05:00
|
|
|
'get_dn',
|
2008-08-08 12:11:29 -05:00
|
|
|
))
|
2008-09-25 21:43:11 -05:00
|
|
|
backend = None
|
2008-09-24 17:19:43 -05:00
|
|
|
methods = None
|
2008-09-24 18:19:34 -05:00
|
|
|
properties = None
|
2008-09-24 18:29:15 -05:00
|
|
|
params = None
|
2008-09-24 20:44:53 -05:00
|
|
|
primary_key = None
|
2008-09-24 22:27:40 -05:00
|
|
|
params_minus_pk = None
|
2008-09-25 21:43:11 -05:00
|
|
|
|
|
|
|
# Can override in subclasses:
|
|
|
|
backend_name = None
|
2008-09-21 19:37:01 -05:00
|
|
|
takes_params = tuple()
|
|
|
|
|
2008-09-21 16:50:56 -05:00
|
|
|
def set_api(self, api):
|
|
|
|
super(Object, self).set_api(api)
|
2008-09-24 20:04:10 -05:00
|
|
|
self.methods = plugable.NameSpace(
|
|
|
|
self.__get_attrs('Method'), sort=False
|
|
|
|
)
|
|
|
|
self.properties = plugable.NameSpace(
|
|
|
|
self.__get_attrs('Property'), sort=False
|
|
|
|
)
|
2008-09-24 18:29:15 -05:00
|
|
|
self.params = plugable.NameSpace(
|
2008-09-24 20:04:10 -05:00
|
|
|
self.__get_params(), sort=False
|
2008-09-24 18:29:15 -05:00
|
|
|
)
|
2008-09-24 20:44:53 -05:00
|
|
|
pkeys = filter(lambda p: p.primary_key, self.params())
|
|
|
|
if len(pkeys) > 1:
|
|
|
|
raise ValueError(
|
|
|
|
'%s (Object) has multiple primary keys: %s' % (
|
|
|
|
self.name,
|
|
|
|
', '.join(p.name for p in pkeys),
|
|
|
|
)
|
|
|
|
)
|
|
|
|
if len(pkeys) == 1:
|
|
|
|
self.primary_key = pkeys[0]
|
2008-09-24 22:27:40 -05:00
|
|
|
self.params_minus_pk = plugable.NameSpace(
|
|
|
|
filter(lambda p: not p.primary_key, self.params()), sort=False
|
|
|
|
)
|
2008-08-05 21:00:18 -05:00
|
|
|
|
2008-09-25 21:43:11 -05:00
|
|
|
if 'Backend' in self.api and self.backend_name in self.api.Backend:
|
|
|
|
self.backend = self.api.Backend[self.backend_name]
|
|
|
|
|
2008-10-14 00:26:24 -05:00
|
|
|
def get_dn(self, primary_key):
|
|
|
|
"""
|
|
|
|
Construct an LDAP DN from a primary_key.
|
|
|
|
"""
|
|
|
|
raise NotImplementedError('%s.get_dn()' % self.name)
|
|
|
|
|
2008-09-24 20:04:10 -05:00
|
|
|
def __get_attrs(self, name):
|
2008-09-25 21:43:11 -05:00
|
|
|
if name not in self.api:
|
|
|
|
return
|
|
|
|
namespace = self.api[name]
|
2008-08-15 14:15:24 -05:00
|
|
|
assert type(namespace) is plugable.NameSpace
|
2008-08-22 16:50:53 -05:00
|
|
|
for proxy in namespace(): # Equivalent to dict.itervalues()
|
2008-08-15 14:15:24 -05:00
|
|
|
if proxy.obj_name == self.name:
|
|
|
|
yield proxy.__clone__('attr_name')
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-09-24 20:04:10 -05:00
|
|
|
def __get_params(self):
|
2008-09-24 19:00:58 -05:00
|
|
|
props = self.properties.__todict__()
|
|
|
|
for spec in self.takes_params:
|
2008-09-24 19:42:38 -05:00
|
|
|
if type(spec) is str:
|
|
|
|
key = spec.rstrip('?*+')
|
|
|
|
else:
|
|
|
|
assert type(spec) is Param
|
|
|
|
key = spec.name
|
|
|
|
if key in props:
|
|
|
|
yield props.pop(key).param
|
2008-09-24 19:00:58 -05:00
|
|
|
else:
|
|
|
|
yield create_param(spec)
|
|
|
|
def get_key(p):
|
|
|
|
if p.param.required:
|
|
|
|
if p.param.default_from is None:
|
|
|
|
return 0
|
|
|
|
return 1
|
|
|
|
return 2
|
|
|
|
for prop in sorted(props.itervalues(), key=get_key):
|
|
|
|
yield prop.param
|
|
|
|
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-08-22 16:27:25 -05:00
|
|
|
class Attribute(plugable.Plugin):
|
2008-10-20 21:28:24 -05:00
|
|
|
"""
|
|
|
|
Base class implementing the attribute-to-object association.
|
|
|
|
|
|
|
|
`Attribute` plugins are associated with an `Object` plugin to group
|
|
|
|
a common set of commands that operate on a common set of parameters.
|
|
|
|
|
|
|
|
The association between attribute and object is done using a simple
|
|
|
|
naming convention: the first part of the plugin class name (up to the
|
|
|
|
first underscore) is the object name, and rest is the attribute name,
|
|
|
|
as this table shows:
|
|
|
|
|
2008-10-21 09:42:52 -05:00
|
|
|
=============== =========== ==============
|
|
|
|
Class name Object name Attribute name
|
|
|
|
=============== =========== ==============
|
|
|
|
noun_verb noun verb
|
|
|
|
user_add user add
|
|
|
|
user_first_name user first_name
|
|
|
|
=============== =========== ==============
|
2008-10-20 21:28:24 -05:00
|
|
|
|
|
|
|
For example:
|
|
|
|
|
|
|
|
>>> class user_add(Attribute):
|
|
|
|
... pass
|
|
|
|
...
|
|
|
|
>>> instance = user_add()
|
|
|
|
>>> instance.obj_name
|
|
|
|
'user'
|
|
|
|
>>> instance.attr_name
|
|
|
|
'add'
|
|
|
|
|
|
|
|
In practice the `Attribute` class is not used directly, but rather is
|
|
|
|
only the base class for the `Method` and `Property` classes. Also see
|
|
|
|
the `Object` class.
|
|
|
|
"""
|
2008-08-11 16:14:07 -05:00
|
|
|
__public__ = frozenset((
|
|
|
|
'obj',
|
|
|
|
'obj_name',
|
|
|
|
))
|
2008-08-08 12:11:29 -05:00
|
|
|
__obj = None
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-08-08 12:11:29 -05:00
|
|
|
def __init__(self):
|
2008-08-22 17:49:56 -05:00
|
|
|
m = re.match(
|
|
|
|
'^([a-z][a-z0-9]+)_([a-z][a-z0-9]+)$',
|
|
|
|
self.__class__.__name__
|
|
|
|
)
|
2008-08-08 16:40:03 -05:00
|
|
|
assert m
|
|
|
|
self.__obj_name = m.group(1)
|
|
|
|
self.__attr_name = m.group(2)
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-08-08 12:11:29 -05:00
|
|
|
def __get_obj_name(self):
|
2008-08-08 16:40:03 -05:00
|
|
|
return self.__obj_name
|
2008-08-08 12:11:29 -05:00
|
|
|
obj_name = property(__get_obj_name)
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-08-08 12:11:29 -05:00
|
|
|
def __get_attr_name(self):
|
2008-08-08 16:40:03 -05:00
|
|
|
return self.__attr_name
|
2008-08-08 12:11:29 -05:00
|
|
|
attr_name = property(__get_attr_name)
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-08-08 12:11:29 -05:00
|
|
|
def __get_obj(self):
|
2008-08-08 16:40:03 -05:00
|
|
|
"""
|
|
|
|
Returns the obj instance this attribute is associated with, or None
|
|
|
|
if no association has been set.
|
|
|
|
"""
|
|
|
|
return self.__obj
|
2008-08-08 12:11:29 -05:00
|
|
|
obj = property(__get_obj)
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-09-21 16:50:56 -05:00
|
|
|
def set_api(self, api):
|
2008-08-22 16:50:53 -05:00
|
|
|
self.__obj = api.Object[self.obj_name]
|
2008-09-21 16:50:56 -05:00
|
|
|
super(Attribute, self).set_api(api)
|
2008-08-05 02:39:50 -05:00
|
|
|
|
|
|
|
|
2008-08-22 16:27:25 -05:00
|
|
|
class Method(Attribute, Command):
|
2008-10-20 20:57:02 -05:00
|
|
|
"""
|
|
|
|
A command with an associated object.
|
|
|
|
|
|
|
|
A `Method` plugin must have a corresponding `Object` plugin. The
|
|
|
|
association between object and method is done through a simple naming
|
|
|
|
convention: the first part of the method name (up to the first under
|
|
|
|
score) is the object name, as the examples in this table show:
|
|
|
|
|
|
|
|
============= =========== ==============
|
|
|
|
Method name Object name Attribute name
|
|
|
|
============= =========== ==============
|
|
|
|
user_add user add
|
|
|
|
noun_verb noun verb
|
2008-10-21 09:42:52 -05:00
|
|
|
door_open_now door open_now
|
2008-10-20 20:57:02 -05:00
|
|
|
============= =========== ==============
|
|
|
|
|
|
|
|
There are three different places a method can be accessed. For example,
|
|
|
|
say you created a `Method` plugin and its corresponding `Object` plugin
|
|
|
|
like this:
|
|
|
|
|
|
|
|
>>> api = plugable.API(Command, Object, Method, Property)
|
|
|
|
>>> class user_add(Method):
|
|
|
|
... def run(self):
|
|
|
|
... return 'Added the user!'
|
|
|
|
...
|
|
|
|
>>> class user(Object):
|
|
|
|
... pass
|
|
|
|
...
|
|
|
|
>>> api.register(user_add)
|
|
|
|
>>> api.register(user)
|
|
|
|
>>> api.finalize()
|
|
|
|
|
|
|
|
First, the ``user_add`` plugin can be accessed through the ``api.Method``
|
|
|
|
namespace:
|
|
|
|
|
|
|
|
>>> list(api.Method)
|
|
|
|
['user_add']
|
|
|
|
>>> api.Method.user_add() # Will call user_add.run()
|
|
|
|
'Added the user!'
|
|
|
|
|
|
|
|
Second, because `Method` is a subclass of `Command`, the ``user_add``
|
|
|
|
plugin can also be accessed through the ``api.Command`` namespace:
|
|
|
|
|
|
|
|
>>> list(api.Command)
|
|
|
|
['user_add']
|
|
|
|
>>> api.Command.user_add() # Will call user_add.run()
|
|
|
|
'Added the user!'
|
|
|
|
|
|
|
|
And third, ``user_add`` can be accessed as an attribute on the ``user``
|
|
|
|
`Object`:
|
|
|
|
|
|
|
|
>>> list(api.Object)
|
|
|
|
['user']
|
|
|
|
>>> list(api.Object.user.methods)
|
|
|
|
['add']
|
|
|
|
>>> api.Object.user.methods.add() # Will call user_add.run()
|
|
|
|
'Added the user!'
|
|
|
|
|
|
|
|
The `Attribute` base class implements the naming convention for the
|
|
|
|
attribute-to-object association. Also see the `Object` and the
|
|
|
|
`Property` classes.
|
|
|
|
"""
|
2008-08-22 16:27:25 -05:00
|
|
|
__public__ = Attribute.__public__.union(Command.__public__)
|
2008-08-05 02:39:50 -05:00
|
|
|
|
2008-09-09 21:02:26 -05:00
|
|
|
def __init__(self):
|
|
|
|
Attribute.__init__(self)
|
|
|
|
Command.__init__(self)
|
|
|
|
|
2008-08-11 16:14:07 -05:00
|
|
|
|
2008-09-02 15:16:34 -05:00
|
|
|
class Property(Attribute):
|
|
|
|
__public__ = frozenset((
|
|
|
|
'rules',
|
2008-09-22 10:33:32 -05:00
|
|
|
'param',
|
2008-09-02 15:16:34 -05:00
|
|
|
'type',
|
|
|
|
)).union(Attribute.__public__)
|
2008-08-05 16:10:49 -05:00
|
|
|
|
2008-09-02 17:19:39 -05:00
|
|
|
type = ipa_types.Unicode()
|
|
|
|
required = False
|
|
|
|
multivalue = False
|
|
|
|
default = None
|
|
|
|
default_from = None
|
|
|
|
normalize = None
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
super(Property, self).__init__()
|
|
|
|
self.rules = tuple(sorted(
|
|
|
|
self.__rules_iter(),
|
|
|
|
key=lambda f: getattr(f, '__name__'),
|
|
|
|
))
|
2008-09-26 20:30:39 -05:00
|
|
|
self.param = Param(self.attr_name,
|
|
|
|
type=self.type,
|
2008-09-09 19:21:40 -05:00
|
|
|
doc=self.doc,
|
2008-09-02 17:19:39 -05:00
|
|
|
required=self.required,
|
|
|
|
multivalue=self.multivalue,
|
|
|
|
default=self.default,
|
|
|
|
default_from=self.default_from,
|
|
|
|
rules=self.rules,
|
|
|
|
normalize=self.normalize,
|
|
|
|
)
|
2008-09-02 15:16:34 -05:00
|
|
|
|
|
|
|
def __rules_iter(self):
|
|
|
|
"""
|
|
|
|
Iterates through the attributes in this instance to retrieve the
|
|
|
|
methods implementing validation rules.
|
|
|
|
"""
|
|
|
|
for name in dir(self.__class__):
|
|
|
|
if name.startswith('_'):
|
|
|
|
continue
|
|
|
|
base_attr = getattr(self.__class__, name)
|
|
|
|
if is_rule(base_attr):
|
|
|
|
attr = getattr(self, name)
|
|
|
|
if is_rule(attr):
|
|
|
|
yield attr
|
2008-09-03 22:34:16 -05:00
|
|
|
|
|
|
|
|
|
|
|
class Application(Command):
|
|
|
|
"""
|
|
|
|
Base class for commands register by an external application.
|
|
|
|
|
|
|
|
Special commands that only apply to a particular application built atop
|
|
|
|
`ipalib` should subclass from ``Application``.
|
|
|
|
|
2008-09-04 03:39:27 -05:00
|
|
|
Because ``Application`` subclasses from `Command`, plugins that subclass
|
2008-09-03 22:34:16 -05:00
|
|
|
from ``Application`` with be available in both the ``api.Command`` and
|
|
|
|
``api.Application`` namespaces.
|
|
|
|
"""
|
|
|
|
|
|
|
|
__public__ = frozenset((
|
|
|
|
'application',
|
2008-09-03 23:39:01 -05:00
|
|
|
'set_application'
|
2008-09-03 22:34:16 -05:00
|
|
|
)).union(Command.__public__)
|
|
|
|
__application = None
|
|
|
|
|
|
|
|
def __get_application(self):
|
|
|
|
"""
|
|
|
|
Returns external ``application`` object.
|
|
|
|
"""
|
|
|
|
return self.__application
|
2008-09-03 23:39:01 -05:00
|
|
|
application = property(__get_application)
|
|
|
|
|
|
|
|
def set_application(self, application):
|
2008-09-03 22:34:16 -05:00
|
|
|
"""
|
|
|
|
Sets the external application object to ``application``.
|
|
|
|
"""
|
|
|
|
if self.__application is not None:
|
|
|
|
raise AttributeError(
|
|
|
|
'%s.application can only be set once' % self.name
|
|
|
|
)
|
|
|
|
if application is None:
|
|
|
|
raise TypeError(
|
|
|
|
'%s.application cannot be None' % self.name
|
|
|
|
)
|
|
|
|
object.__setattr__(self, '_Application__application', application)
|
|
|
|
assert self.application is application
|