Rename AttrDef to wsattr, and introduce wsproperty

wsme/tests/test_types.py

@@ -0,0 +1,23 @@
+import unittest
+from wsme import types
+
+class TestTypes(unittest.TestCase):
+    def test_flat_type(self):
+        class Flat(object):
+            aint = int
+            astr = str
+            auni = unicode
+            afloat = float
+
+        types.register_type(Flat)
+
+        assert len(Flat._wsme_attributes) == 4
+
+    def test_private_attr(self):
+        class WithPrivateAttrs(object):
+            _private = 12
+        
+        types.register_type(WithPrivateAttrs)
+
+        assert len(WithPrivateAttrs._wsme_attributes) == 0
+

wsme/types.py

@@ -12,18 +12,19 @@ native_types = pod_types + dt_types + extra_types
 
 structured_types = []
 
-def mandatory(datatype):
-    if isinstance(datatype, AttrDef):
-        datatype.mandatory = True
-        return datatype
-    return AttrDef(datatype, True)
+
+class wsproperty(property):
+    def __init__(self, datatype, fget, fset=None,
+                 mandatory=False, doc=None):
+        property.__init__(self, fget, fset, doc)
+        self.mandatory = mandatory
 
 
-class AttrDef(object):
-    def __init__(self, datatype, mandatory=False, default=None):
+class wsattr(object):
+    def __init__(self, datatype, mandatory=False):
         self.datatype = datatype
         self.mandatory = mandatory
-        self.default = None
+
 
 def inspect_class(class_):
     attributes = []
@@ -36,12 +37,12 @@ def inspect_class(class_):
             continue
         if inspect.ismethod(attr):
             continue
-        if not isinstance(attr, AttrDef):
-            attrdef = AttrDef(attr)
+        if not isinstance(attr, wsattr):
+            attrdef = wsattr(attr)
         else:
             attrdef = attr
 
-        attributes.append((name, AttrDef))
+        attributes.append((name, wsattr))
     return attributes
 
 def register_type(class_):