202: Started work on type classes in ipa_types module; added corresponding unit tests

1 files changed, 103 insertions, 0 deletions

diff --git a/ipalib/ipa_types.py b/ipalib/ipa_types.py
new file mode 100644
index 000000000..9a34a6b63
--- /dev/null
+++ b/ipalib/ipa_types.py
@@ -0,0 +1,103 @@
+# 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
+
+"""
+Type system for coercing and normalizing input values.
+"""
+
+from plugable import ReadOnly, lock
+import errors
+
+
+class Type(ReadOnly):
+    """
+    Base class for all IPA types.
+    """
+
+    type = None # Override in base class
+
+    def convert(self, value):
+		return self.type(value)
+
+    def __get_name(self):
+        """
+        Convenience property to return the class name.
+        """
+        return self.__class__.__name__
+    name = property(__get_name)
+
+
+class Int(Type):
+    type = int
+
+    def __init__(self, min_value=None, max_value=None):
+        args = (min_value, max_value)
+        for arg in args:
+            if not (arg is None or type(arg) is int):
+                raise TypeError('Must be an int or None: %r' % arg)
+        if None not in args and min_value >= max_value:
+            raise ValueError(
+                'min_value not less than max_value: %r, %r' % (
+                    min_value, max_value
+                )
+            )
+        self.min_value = min_value
+        self.max_value = max_value
+        lock(self)
+
+    def validate(self, value):
+        if type(value) is not self.type:
+            return 'Must be an integer'
+        if self.min_value is not None and value < self.min_value:
+            return 'Cannot be smaller than %d' % self.min_value
+        if self.max_value is not None and value > self.max_value:
+            return 'Cannot be larger than %d' % self.max_value
+
+
+def check_min_max(min_name, min_value, max_name, max_value):
+    assert type(min_name) is str, 'min_name must be an str'
+    assert type(max_name) is str, 'max_name must be an str'
+    for (name, value) in [(min_name, min_value), (max_name, max_value)]:
+        if not (value is None or type(value) is int):
+            raise TypeError(
+                '%s must be an int or None, got: %r' % (name, value)
+            )
+#    if None not in (min_value, max_value) and min_value >= max_value:
+#        raise ValueError(
+#            'min_value not less than max_value: %r, %r' % (
+#                min_value, max_value
+#            )
+#        )
+
+
+class Unicode(Type):
+    def __init__(self, min_length=None, max_length=None, pattern=None):
+        integers = (min_length, max_length)
+        for i in integers:
+            if not (i is None or type(i) is int):
+                raise TypeError('Must be an int or None: %r' % i)
+        if None not in integers and min_value >= max_value:
+            raise ValueError(
+                'min_value not less than max_value: %r, %r' % (
+                    min_value, max_value
+                )
+            )
+        self.min_length = min_length
+        self.max_length = max_length
+        self.pattern = pattern