New Param: changed kwargs class attribute to a tuple so the subclass interface is simpler

This commit is contained in:
Jason Gerard DeRose 2008-12-11 22:39:50 -07:00
parent 64ae4bc986
commit 079721da2c
2 changed files with 75 additions and 65 deletions

View File

@ -21,6 +21,7 @@
Parameter system for command plugins.
"""
from types import NoneType
from plugable import ReadOnly, lock, check_name
from constants import NULLS, TYPE_ERROR, CALLABLE_ERROR
@ -178,44 +179,46 @@ class DefaultFrom(ReadOnly):
class Param(ReadOnly):
"""
Base class for all IPA types.
Base class for all parameters.
"""
__kwargs = dict(
cli_name=(str, None),
doc=(str, ''),
required=(bool, True),
multivalue=(bool, False),
primary_key=(bool, False),
normalizer=(callable, None),
default=(None, None),
default_from=(callable, None),
flags=(frozenset, frozenset()),
kwargs = (
('cli_name', str, None),
('doc', str, ''),
('required', bool, True),
('multivalue', bool, False),
('primary_key', bool, False),
('normalizer', callable, None),
('default_from', callable, None),
('flags', frozenset, frozenset()),
# The 'default' kwarg gets appended in Param.__init__():
# ('default', self.type, None),
)
def __init__(self, name, kwargs, **override):
# This is a dummy type so that most of the functionality of Param can be
# unit tested directly without always creating a subclass; however, real
# (direct) subclasses should *always* override this class attribute:
type = NoneType # This isn't very useful in the real world!
def __init__(self, name, **kw):
assert type(self.type) is type
self.param_spec = name
self.__override = dict(override)
if not ('required' in override or 'multivalue' in override):
self.__kw = dict(kw)
if not ('required' in kw or 'multivalue' in kw):
(name, kw_from_spec) = parse_param_spec(name)
override.update(kw_from_spec)
kw.update(kw_from_spec)
self.name = check_name(name)
if 'cli_name' not in override:
override['cli_name'] = self.name
df = override.get('default_from', None)
if kw.get('cli_name', None) is None:
kw['cli_name'] = self.name
df = kw.get('default_from', None)
if callable(df) and not isinstance(df, DefaultFrom):
override['default_from'] = DefaultFrom(df)
kwargs = dict(kwargs)
assert set(self.__kwargs).intersection(kwargs) == set()
kwargs.update(self.__kwargs)
for (key, (kind, default)) in kwargs.iteritems():
value = override.get(key, default)
if value is None:
if kind is bool:
raise TypeError(
TYPE_ERROR % (key, bool, value, type(value))
)
else:
kw['default_from'] = DefaultFrom(df)
self.__clonekw = kw
self.kwargs += (('default', self.type, None),)
for (key, kind, default) in self.kwargs:
value = kw.get(key, default)
if value is not None:
if (
type(kind) is type and type(value) is not kind
or
@ -275,7 +278,7 @@ class Param(ReadOnly):
"""
Normalize a scalar value.
This method is called once for each value in multivalue.
This method is called once for each value in a multivalue.
"""
if type(value) is not unicode:
return value
@ -308,8 +311,6 @@ class Param(ReadOnly):
)
class Bool(Param):
"""
@ -333,15 +334,27 @@ class Bytes(Param):
"""
type = str
def __init__(self, name, **kw):
kwargs = dict(
minlength=(int, None),
maxlength=(int, None),
length=(int, None),
pattern=(str, None),
)
class Str(Param):
"""
"""
def __init__(self, name, **overrides):
self.type = unicode
super(Str, self).__init__(name, {}, **overrides)
type = unicode
def __init__(self, name, **kw):
super(Str, self).__init__(name, **kw)
def _convert_scalar(self, value, index=None):
if type(value) in (self.type, int, float, bool):

View File

@ -92,6 +92,8 @@ def test_parse_param_spec():
assert f('name?') == ('name', dict(required=False, multivalue=False))
assert f('name*') == ('name', dict(required=False, multivalue=True))
assert f('name+') == ('name', dict(required=True, multivalue=True))
# Make sure other "funny" endings are treated special:
assert f('name^') == ('name^', dict(required=True, multivalue=False))
class test_Param(ClassChecker):
@ -105,72 +107,67 @@ class test_Param(ClassChecker):
Test the `ipalib.parameter.Param.__init__` method.
"""
name = 'my_param'
o = self.cls(name, {})
o = self.cls(name)
assert o.name is name
assert o.__islocked__() is True
# Test default values:
assert o.name is name
# Test default kwarg values:
assert o.cli_name is name
assert o.doc == ''
assert o.required is True
assert o.multivalue is False
assert o.primary_key is False
assert o.normalizer is None
assert o.default is None
#assert o.default is None
assert o.default_from is None
assert o.flags == frozenset()
# Test that ValueError is raised when a kwarg from a subclass
# conflicts with an attribute:
kwarg = dict(convert=(callable, None))
e = raises(ValueError, self.cls, name, kwarg)
assert str(e) == "kwarg 'convert' conflicts with attribute on Param"
class Subclass(self.cls):
pass
e = raises(ValueError, Subclass, name, kwarg)
kwargs = self.cls.kwargs + (
('convert', callable, None),
)
e = raises(ValueError, Subclass, name)
assert str(e) == "kwarg 'convert' conflicts with attribute on Subclass"
# Test type validation of keyword arguments:
kwargs = dict(
extra1=(bool, True),
extra2=(str, 'Hello'),
extra3=((int, float), 42),
extra4=(callable, lambda whatever: whatever + 7),
)
# Note: we don't accept None if kind is bool:
e = raises(TypeError, self.cls, 'my_param', kwargs, extra1=None)
assert str(e) == TYPE_ERROR % ('extra1', bool, None, type(None))
for (key, (kind, default)) in kwargs.items():
o = self.cls('my_param', kwargs)
class Subclass(self.cls):
kwargs = self.cls.kwargs + (
('extra1', bool, True),
('extra2', str, 'Hello'),
('extra3', (int, float), 42),
('extra4', callable, lambda whatever: whatever + 7),
)
o = Subclass('my_param') # Test with no **kw:
for (key, kind, default) in o.kwargs:
# Test with a type invalid for all:
value = object()
overrides = {key: value}
e = raises(TypeError, self.cls, 'my_param', kwargs, **overrides)
kw = {key: value}
e = raises(TypeError, Subclass, 'my_param', **kw)
if kind is callable:
assert str(e) == CALLABLE_ERROR % (key, value, type(value))
else:
assert str(e) == TYPE_ERROR % (key, kind, value, type(value))
if kind is bool: # See note above
continue
# Test with None:
overrides = {key: None}
o = self.cls('my_param', kwargs, **overrides)
kw = {key: None}
Subclass('my_param', **kw)
def test_convert_scalar(self):
"""
Test the `ipalib.parameter.Param._convert_scalar` method.
"""
o = self.cls('my_param', {})
o = self.cls('my_param')
e = raises(NotImplementedError, o._convert_scalar, 'some value')
assert str(e) == 'Param._convert_scalar()'
class Subclass(self.cls):
pass
o = Subclass('my_param', {})
o = Subclass('my_param')
e = raises(NotImplementedError, o._convert_scalar, 'some value')
assert str(e) == 'Subclass._convert_scalar()'
class test_Str(ClassChecker):
"""
Test the `ipalib.parameter.Str` class.