[MERGE] Forward-port saas-5 up to f9bcd67
[odoo/odoo.git] / openerp / modules / loading.py
index 0299945..61e573d 100644 (file)
@@ -29,6 +29,7 @@ import logging
 import os
 import sys
 import threading
+import time
 
 import openerp
 import openerp.modules.db
@@ -51,7 +52,7 @@ _test_logger = logging.getLogger('openerp.tests')
 def load_module_graph(cr, graph, status=None, perform_checks=True, skip_modules=None, report=None):
     """Migrates+Updates or Installs all module nodes from ``graph``
        :param graph: graph of module nodes to load
-       :param status: status dictionary for keeping track of progress
+       :param status: deprecated parameter, unused, left to avoid changing signature in 8.0
        :param perform_checks: whether module descriptors should be checked for validity (prints warnings
                               for same cases)
        :param skip_modules: optional list of module names (packages) which have previously been loaded and can be skipped
@@ -110,7 +111,7 @@ def load_module_graph(cr, graph, status=None, perform_checks=True, skip_modules=
             if kind in ('demo', 'test'):
                 threading.currentThread().testing = True
             for filename in _get_files_of_kind(kind):
-                _logger.info("module %s: loading %s", module_name, filename)
+                _logger.info("loading %s/%s", module_name, filename)
                 noupdate = False
                 if kind in ('demo', 'demo_xml') or (filename.endswith('.csv') and kind in ('init', 'init_xml')):
                     noupdate = True
@@ -119,9 +120,6 @@ def load_module_graph(cr, graph, status=None, perform_checks=True, skip_modules=
             if kind in ('demo', 'test'):
                 threading.currentThread().testing = False
 
-    if status is None:
-        status = {}
-
     processed_modules = []
     loaded_modules = []
     registry = openerp.registry(cr.dbname)
@@ -137,6 +135,9 @@ def load_module_graph(cr, graph, status=None, perform_checks=True, skip_modules=
         registry.fields_by_model.setdefault(field['model'], []).append(field)
 
     # register, instantiate and initialize models for each modules
+    t0 = time.time()
+    t0_sql = openerp.sql_db.sql_counter
+
     for index, package in enumerate(graph):
         module_name = package.name
         module_id = package.id
@@ -144,16 +145,22 @@ def load_module_graph(cr, graph, status=None, perform_checks=True, skip_modules=
         if skip_modules and module_name in skip_modules:
             continue
 
-        _logger.debug('module %s: loading objects', package.name)
         migrations.migrate_module(package, 'pre')
         load_openerp_module(package.name)
 
+        new_install = package.installed_version is None
+        if new_install:
+            py_module = sys.modules['openerp.addons.%s' % (module_name,)]
+            pre_init = package.info.get('pre_init_hook')
+            if pre_init:
+                getattr(py_module, pre_init)(cr)
+
         models = registry.load(cr, package)
 
         loaded_modules.append(package.name)
         if hasattr(package, 'init') or hasattr(package, 'update') or package.state in ('to install', 'to upgrade'):
+            registry.setup_models(cr, partial=True)
             init_module_models(cr, package.name, models)
-        status['progress'] = float(index) / len(graph)
 
         # Can't put this line out of the loop: ir.module.module will be
         # registered by init_module_models() above.
@@ -175,12 +182,17 @@ def load_module_graph(cr, graph, status=None, perform_checks=True, skip_modules=
             _load_data(cr, module_name, idref, mode, kind='data')
             has_demo = hasattr(package, 'demo') or (package.dbdemo and package.state != 'installed')
             if has_demo:
-                status['progress'] = (index + 0.75) / len(graph)
                 _load_data(cr, module_name, idref, mode, kind='demo')
                 cr.execute('update ir_module_module set demo=%s where id=%s', (True, module_id))
+                modobj.invalidate_cache(cr, SUPERUSER_ID, ['demo'], [module_id])
 
             migrations.migrate_module(package, 'post')
 
+            if new_install:
+                post_init = package.info.get('post_init_hook')
+                if post_init:
+                    getattr(py_module, post_init)(cr, registry)
+
             registry._init_modules.add(package.name)
             # validate all the views at a whole
             registry['ir.ui.view']._validate_module_views(cr, SUPERUSER_ID, module_name)
@@ -213,10 +225,14 @@ def load_module_graph(cr, graph, status=None, perform_checks=True, skip_modules=
         registry._init_modules.add(package.name)
         cr.commit()
 
+    registry.setup_models(cr, partial=True)
+
+    _logger.log(25, "%s modules loaded in %.2fs, %s queries", len(graph), time.time() - t0, openerp.sql_db.sql_counter - t0_sql)
+
     # The query won't be valid for models created later (i.e. custom model
     # created after the registry has been loaded), so empty its result.
     registry.fields_by_model = None
-    
+
     cr.commit()
 
     return loaded_modules, processed_modules
@@ -251,9 +267,6 @@ def load_marked_modules(cr, graph, states, force, progressdict, report, loaded_m
     return processed_modules
 
 def load_modules(db, force_demo=False, status=None, update_module=False):
-    # TODO status['progress'] reporting is broken: used twice (and reset each
-    # time to zero) in load_module_graph, not fine-grained enough.
-    # It should be a method exposed by the registry.
     initialize_sys_path()
 
     force = []
@@ -316,6 +329,7 @@ def load_modules(db, force_demo=False, status=None, update_module=False):
                     modobj.button_upgrade(cr, SUPERUSER_ID, ids)
 
             cr.execute("update ir_module_module set state=%s where name=%s", ('installed', 'base'))
+            modobj.invalidate_cache(cr, SUPERUSER_ID, ['state'])
 
 
         # STEP 3: Load marked modules (skipping base which was done in STEP 1)
@@ -347,6 +361,7 @@ def load_modules(db, force_demo=False, status=None, update_module=False):
         cr.execute('select model from ir_model where state=%s', ('manual',))
         for model in cr.dictfetchall():
             registry['ir.model'].instanciate(cr, SUPERUSER_ID, model['model'], {})
+        registry.setup_models(cr)
 
         # STEP 4: Finish and cleanup installations
         if processed_modules:
@@ -402,10 +417,17 @@ def load_modules(db, force_demo=False, status=None, update_module=False):
         if update_module:
             # Remove records referenced from ir_model_data for modules to be
             # removed (and removed the references from ir_model_data).
-            cr.execute("SELECT id FROM ir_module_module WHERE state=%s", ('to remove',))
-            mod_ids_to_remove = [x[0] for x in cr.fetchall()]
-            if mod_ids_to_remove:
-                registry['ir.module.module'].module_uninstall(cr, SUPERUSER_ID, mod_ids_to_remove)
+            cr.execute("SELECT name, id FROM ir_module_module WHERE state=%s", ('to remove',))
+            modules_to_remove = dict(cr.fetchall())
+            if modules_to_remove:
+                pkgs = reversed([p for p in graph if p.name in modules_to_remove])
+                for pkg in pkgs:
+                    uninstall_hook = pkg.info.get('uninstall_hook')
+                    if uninstall_hook:
+                        py_module = sys.modules['openerp.addons.%s' % (pkg.name,)]
+                        getattr(py_module, uninstall_hook)(cr, registry)
+
+                registry['ir.module.module'].module_uninstall(cr, SUPERUSER_ID, modules_to_remove.values())
                 # Recursive reload, should only happen once, because there should be no
                 # modules to remove next time
                 cr.commit()
@@ -433,10 +455,14 @@ def load_modules(db, force_demo=False, status=None, update_module=False):
 
         # STEP 9: Run the post-install tests
         cr.commit()
+
+        t0 = time.time()
+        t0_sql = openerp.sql_db.sql_counter
         if openerp.tools.config['test_enable']:
             cr.execute("SELECT name FROM ir_module_module WHERE state='installed'")
             for module_name in cr.fetchall():
                 report.record_result(openerp.modules.module.run_unit_tests(module_name[0], cr.dbname, position=runs_post_install))
+            _logger.log(25, "All post-tested in %.2fs, %s queries", time.time() - t0, openerp.sql_db.sql_counter - t0_sql)
     finally:
         cr.close()