Add a new get_class_childs to Model class and a child_classes method to LeObject

Now lefactory populate a LeObject attribute named _child_classes

lodel/editorial_model/model.py

@@ -96,7 +96,19 @@ class EditorialModel(object):
                                         uid)
         except KeyError:
             raise EditorialModelException("EmClass not found : '%s'" % uid)
-             
+    
+    ##@brief EmClass child list accessor
+    #@param uid str : the EmClass uid
+    #@return a set of EmClass
+    def get_class_childs(self, uid):
+        res = list()
+        cur = self.classes(uid)
+        for cls in self.classes():
+            if cur in cls.parents_recc:
+                res.append(cls)
+        return set(res)
+
+
     ##@brief EmGroup getter
     # @param uid None | str : give this argument to get a specific EmGroup
     # @return if uid is given returns an EmGroup else returns an EmGroup iterator

lodel/leapi/lefactory.py

@@ -132,6 +132,7 @@ class {clsname}({parents}):
     _ro_datasource = None
     _rw_datasource = None
     _datasource_name = {datasource_name}
+    _child_classes = None
 
 """.format(
     clsname = LeObject.name2objname(em_class.uid),
@@ -143,9 +144,11 @@ class {clsname}({parents}):
         res += em_cls_code
         # Dyncode fields bootstrap instructions
         bootstrap += """{classname}._set__fields({fields})
+{classname}._child_classes = {child_classes}
 """.format(
     classname = LeObject.name2objname(em_class.uid),
     fields = '{' + (', '.join(['\n\t%s: %s' % (repr(emfield.uid),data_handler_constructor(emfield)) for emfield in em_class.fields()])) + '}',
+    child_classes = '(' + (', '.join([ LeObject.name2objname(emcls.uid) for emcls in model.get_class_childs(em_class.uid)]))+ ')',
 )
     bootstrap += "\n"
     return res, set(imports), bootstrap

lodel/leapi/leobject.py

@@ -57,8 +57,10 @@ class LeObject(object):
     _uid = None 
     ##@brief Read only datasource ( see @ref lodel2_datasources )
     _ro_datasource = None
-    ##@breif Read & write datasource ( see @ref lodel2_datasources )
+    ##@brief Read & write datasource ( see @ref lodel2_datasources )
     _rw_datasource = None
+    ##@brief Store the list of child classes
+    _child_classes = None
 
     def __new__(cls, **kwargs):
         
@@ -201,6 +203,13 @@ class LeObject(object):
                 res.append(cur)
         return res
     
+    ##@brief Return a tuple a child classes
+    #@return a tuple of child classes
+    @classmethod
+    def child_classes(cls):
+        return copy.copy(cls._child_classes)
+        
+
     ##@brief Return the parent class that is the "source" of uid
     #
     #The method goal is to return the parent class that defines UID.