New graphviz backend + some methods added to some components

EditorialModel/backend/graphviz.py

@@ -0,0 +1,126 @@
+# -*- coding: utf-8 -*-
+
+import datetime
+from EditorialModel.classtypes import EmClassType
+from EditorialModel.fieldgroups import EmFieldGroup
+from EditorialModel.types import EmType
+from Lodel.utils.mlstring import MlString
+
+class EmBackendGraphviz(object):
+
+    ## @brief Constructor
+    # @param dot_fname str : The filename where we want to save the dot repr of the EM
+    def __init__(self, dot_fname):
+        self.edges = ""
+        self.dot_fname = dot_fname
+        #with open(dot_file, 'w+') as dot_fp:
+    
+    ## @brief Not implementend
+    # @warning Not implemented
+    def load(self):
+        raise NotImplementedError(self.__class__.__name__+' cannot load an EM')
+
+    ## @brief Save an EM in a dot file
+    # @param em model : The EM to save
+    # @warning hardcoded classtype
+    def save(self, em):
+        self.edges = ""
+        with open(self.dot_fname, 'w') as dotfp:
+            dotfp.write("digraph G {\n\trankdir = BT\n")
+            
+            dotfp.write('subgraph cluster_classtype {\nstyle="invis"\n')
+            for ct in [ 'entity', 'entry', 'person' ]:
+                dotfp.write('\n\nct%s [ label="classtype %s" shape="tripleoctagon" ]\n'%(ct, ct))
+            dotfp.write("}\n")
+
+            
+            dotfp.write('subgraph cluster_class {\nstyle="invis"\n')
+            for c in em.classes():
+
+                dotfp.write(self._component_node(c, em))
+                cn = c.__class__.__name__
+                cid = self._component_id(c)
+                self.edges += cid+' -> ct%s [ style="dashed" ]\n'%c.classtype
+            dotfp.write("}\n")
+
+            #dotfp.write('subgraph cluster_fieldgroup {\nstyle="invis"\n')
+            for c in em.components(EmFieldGroup):
+                dotfp.write(self._component_node(c, em))
+                cn = c.__class__.__name__
+                cid = self._component_id(c)
+                self.edges += cid+' -> '+self._component_id(c.em_class)+' [ style="dashed" ]\n'
+            #dotfp.write("}\n")
+
+
+            #dotfp.write('subgraph cluster_type {\nstyle="invis"\n')
+            for c in em.components(EmType):
+                dotfp.write(self._component_node(c, em))
+                cn = c.__class__.__name__
+                cid = self._component_id(c)
+                self.edges += cid+' -> '+self._component_id(c.em_class)+' [ style="dotted" ]\n'
+                for fg in c.fieldgroups():
+                    self.edges += cid+' -> '+self._component_id(fg)+' [ style="dashed" ]\n'
+                for nat in c.superiors():
+                    self.edges += cid+' -> '+self._component_id(c.superiors()[nat])+' [ label="%s" color="green" ]'%nat
+            #dotfp.write("}\n")
+
+            dotfp.write(self.edges)
+
+            dotfp.write("\n}")
+        pass
+
+    @staticmethod
+    def _component_id(c):
+        return 'emcomp%d'%c.uid
+
+    def _component_node(self, c, em):
+        #ret = 'emcomp%d '%c.uid
+        ret = "\t"+EmBackendGraphviz._component_id(c)
+        cn = c.__class__.__name__
+        rel_field = ""
+        if cn == 'EmClass':
+            ret += '[ label="%s", shape="%s" ]'%(c.name, 'doubleoctagon')
+        elif cn == 'EmType' or cn == 'EmFieldGroup':
+            ret += '[ label="%s %s '%(cn, c.name)
+
+            cntref = 0
+            first = True
+            for f in c.fields():
+                if ((cn == 'EmType' and f.optional) or (cn == 'EmFieldGroup' and not f.optional)) and f.rel_field_id is None:
+                    
+                    if not (f.rel_to_type_id is None):
+                        rel_node_id = '%s%s'%(EmBackendGraphviz._component_id(c), EmBackendGraphviz._component_id(em.component(f.rel_to_type_id)))
+
+                        rel_node = '\t%s [ label="rel_to_type'%rel_node_id
+
+                        if len(f.rel_to_type_fields()) > 0:
+                            #rel_node += '| {'
+                            first = True
+                            for rf in f.rel_to_type_fields():
+                                rel_node += ' | '
+                                if first:
+                                    rel_node += '{ '
+                                    first = False
+                                rel_node += rf.name
+                        rel_node += '}" shape="record" style="dashed"]\n'
+
+                        rel_field += rel_node
+
+                        ref_node = EmBackendGraphviz._component_id(em.component(f.rel_to_type_id))
+                        self.edges += '%s:f%d -> %s [ color="purple" ]\n'%(EmBackendGraphviz._component_id(c), cntref, rel_node_id)
+                        self.edges += '%s -> %s [color="purple"]\n'%(rel_node_id, ref_node)
+
+                    ret += '|'
+                    if first:
+                        ret += ' { '
+                        first = False
+                    if not (f.rel_to_type_id is None):
+                        ret += '<f%d> '%cntref
+                        cntref += 1
+                    ret += f.name
+            ret += '}" shape="record" color="%s" ]'%('blue' if cn == 'EmType' else 'red')
+        else:
+            return ""
+        ret +="\n"+rel_field
+        return ret
+            

EditorialModel/backend/json_backend.py

@@ -69,5 +69,5 @@ class EmBackendJson(object):
     #
     # @return bool
     # @todo à implémenter
-    def save(self):
+    def save(self, em):
         return True

EditorialModel/fieldgroups.py

@@ -20,6 +20,10 @@ class EmFieldGroup(EmComponent):
         self.check_type('class_id', int)
         super(EmFieldGroup, self).__init__(model=model, uid=uid, name=name, string=string, help_text=help_text, date_update=date_update, date_create=date_create, rank=rank)
 
+    @property
+    def em_class(self):
+        return self.model.component(self.class_id)
+
     ## Check if the EmFieldGroup is valid
     # @throw EmComponentCheckError if fails
     def check(self):

EditorialModel/fields.py

@@ -48,6 +48,14 @@ class EmField(EmComponent):
 
         super(EmField, self).__init__(model=model, uid=uid, name=name, string=string, help_text=help_text, date_update=date_update, date_create=date_create, rank=rank)
 
+    ## @brief Return the list of relation fields for a rel_to_type
+    # @return None if the field is not a rel_to_type else return a list of EmField
+    def rel_to_type_fields(self):
+        if not self.rel_to_type_id:
+            return None
+        
+        return [ f for f in self.model.components(EmField) if f.rel_field_id == self.uid ]
+
     ## Check if the EmField is valid
     # @return True if valid False if not
     def check(self):

EditorialModel/model.py

@@ -91,7 +91,7 @@ class Model(object):
 
     ## Saves data using the current backend
     def save(self):
-        return self.backend.save()
+        return self.backend.save(self)
 
     ## Given a EmComponent child class return a list of instances
     # @param cls EmComponent : A python class