dyncode and init_datasources in Commands

management/commands/generate_dyncode.py

@@ -7,13 +7,21 @@ class GenerateDyncode(Command):
     option_list = (
         Option('--model', '-m', dest='model_file'),
         Option('--translator', '-t', dest='translator'),
+        Option('--filename', '-f', dest='filename')
     )
     
     ## @brief Generates the dynamic code corresponding to an editorial model
     # @param model_file str : file path corresponding to a given editorial model
     # @param translator
     # @return module
-    def run(self, model_file, translator):
+    def run(self, model_file, translator, filename):
         dyncode = generate_dyncode(model_file, translator)        
         print("Dyncode successfully generated for the editorial model from the file %s" % model_file)
+        
+        if filename is not None:
+            with open(filename, 'w+') as out_fd:
+                out_fd.write(dyncode)
+                out_fd.close()
+                print("Dynamic leapi code written in %s", filename)
+                
         return dyncode

management/commands/init_datasources.py

@@ -0,0 +1,17 @@
+from flask_script import Command, Option
+
+from management.utils import init_all_datasources
+
+class InitDataSources(Command):
+
+    option_list = (
+        Option('--dyncode', '-m', dest='dyncode'),
+    )
+    
+    ## @brief Generates the dynamic code corresponding to an editorial model
+    # @param model_file str : file path corresponding to a given editorial model
+    # @param translator
+    # @return module
+    def run(self, dyncode):
+        init_all_datasources(dyncode)       
+        print("Datasources successfully started")

management/utils.py

@@ -16,3 +16,58 @@ def generate_dyncode(model_file, translator):
     model = EditorialModel.load(translator, filename=model_file)
     dyncode = lefactory.dyncode_from_em(model)
     return dyncode
+
+def init_all_datasources(dyncode):
+    from lodel.plugin.datasource_plugin import DatasourcePlugin
+    from lodel import logger
+
+    ds_cls = dict() # EmClass indexed by rw_datasource
+    for cls in dyncode.dynclasses:
+        ds = cls._datasource_name
+        if ds not in ds_cls:
+            ds_cls[ds] = [cls]
+        else:
+            ds_cls[ds].append(cls)
+    
+    for ds_name in ds_cls:
+        mh = DatasourcePlugin.init_migration_handler(ds_name)
+        #Retrieve plugin_name & ds_identifier for logging purpose
+        plugin_name, ds_identifier = DatasourcePlugin.plugin_name(
+            ds_name, False)
+        try:
+            mh.init_db(ds_cls[ds_name])
+        except Exception as e:
+            msg = "Migration failed for datasource %s(%s.%s) when running \
+init_db method: %s"
+            msg %= (ds_name, plugin_name, ds_identifier, e)
+        logger.info("Database initialisation done for %s(%s.%s)" % (
+            ds_name, plugin_name, ds_identifier))
+
+def list_registered_hooks():
+    from lodel.plugin.hooks import LodelHook
+    from lodel.plugin.plugins import Plugin
+    
+    Plugin.load_all()
+    hlist = LodelHook.hook_list()
+    print("Registered hooks are : ")
+    for name in sorted(hlist.keys()):
+        print("\t- %s is registered by : " % name)
+        for reg_hook in hlist[name]:
+            hook, priority = reg_hook
+            msg = "\t\t- {modname}.{funname} with priority : {priority}"
+            msg = msg.format(
+                modname = hook.__module__,
+                funname = hook.__name__,
+                priority = priority)
+            print(msg)
+        print("\n")
+
+##@brief update plugin's discover cache
+def update_plugin_discover_cache(path_list = None):
+    os.environ['LODEL2_NO_SETTINGS_LOAD'] = 'True'
+    from lodel.plugin.plugins import Plugin
+    res = Plugin.discover(path_list)
+    print("Plugin discover result in %s :\n" % res['path_list'])
+    for pname, pinfos in res['plugins'].items():
+        print("\t- %s %s -> %s" % (
+            pname, pinfos['version'], pinfos['path']))