[IMP] use the openerp namespace.
[odoo/odoo.git] / openerp / addons / base / module / wizard / base_module_upgrade.py
index ca82cfb..bd6b9b7 100644 (file)
@@ -1,8 +1,8 @@
 # -*- coding: utf-8 -*-
 ##############################################################################
 #
-#    OpenERP, Open Source Management Solution
-#    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
+#    OpenERP, Open Source Business Applications
+#    Copyright (C) 2004-2012 OpenERP SA (<http://openerp.com>).
 #
 #    This program is free software: you can redistribute it and/or modify
 #    it under the terms of the GNU Affero General Public License as
@@ -19,9 +19,9 @@
 #
 ##############################################################################
 
-import pooler
-from osv import osv, fields
-from tools.translate import _
+from openerp import pooler
+from openerp.osv import osv, fields
+from openerp.tools.translate import _
 
 class base_module_upgrade(osv.osv_memory):
     """ Module Upgrade """
@@ -30,18 +30,15 @@ class base_module_upgrade(osv.osv_memory):
     _description = "Module Upgrade"
 
     _columns = {
-        'module_info': fields.text('Modules to update',readonly=True),
+        'module_info': fields.text('Modules to Update',readonly=True),
     }
 
     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
-        """ Changes the view dynamically
-         @param self: The object pointer.
-         @param cr: A database cursor
-         @param uid: ID of the user currently logged in
-         @param context: A standard dictionary
-         @return: New arch of view.
-        """
         res = super(base_module_upgrade, self).fields_view_get(cr, uid, view_id=view_id, view_type=view_type, context=context, toolbar=toolbar,submenu=False)
+        if view_type != 'form':
+            return res
+
+        context = {} if context is None else context
         record_id = context and context.get('active_id', False) or False
         active_model = context.get('active_model')
         if (not record_id) or (not active_model):
@@ -49,14 +46,12 @@ class base_module_upgrade(osv.osv_memory):
 
         ids = self.get_module_list(cr, uid, context=context)
         if not ids:
-            res['arch'] = '''<form string="Apply Scheduled Upgrades">
-                                <separator string="System update completed" colspan="4"/>
-                                <label align="0.0" string="The selected modules have been updated / installed !" colspan="4"/>
-                                <label align="0.0" string="We suggest to reload the menu tab to see the new menus (Ctrl+T then Ctrl+R)." colspan="4"/>
-                                 <separator string="" colspan="4"/>
-                                <newline/>
-                                <button special="cancel" string="Close" icon="gtk-cancel"/>
-                                <button name="config" string="Start configuration" type="object" icon="gtk-ok"/>
+            res['arch'] = '''<form string="Upgrade Completed" version="7.0">
+                                <separator string="Upgrade Completed" colspan="4"/>
+                                <footer>
+                                    <button name="config" string="Start Configuration" type="object" class="oe_highlight"/> or
+                                    <button special="cancel" string="Close" class="oe_link"/>
+                                </footer>
                              </form>'''
 
         return res
@@ -68,45 +63,39 @@ class base_module_upgrade(osv.osv_memory):
         return ids
 
     def default_get(self, cr, uid, fields, context=None):
-        """
-        This function checks for precondition before wizard executes
-        @param self: The object pointer
-        @param cr: the current row, from the database cursor,
-        @param uid: the current user’s ID for security checks,
-        @param fields: List of fields for default value
-        @param context: A standard dictionary for contextual values
-        """
         mod_obj = self.pool.get('ir.module.module')
         ids = self.get_module_list(cr, uid, context=context)
         res = mod_obj.read(cr, uid, ids, ['name','state'], context)
         return {'module_info': '\n'.join(map(lambda x: x['name']+' : '+x['state'], res))}
 
     def upgrade_module(self, cr, uid, ids, context=None):
-        mod_obj = self.pool.get('ir.module.module')
-        ids = mod_obj.search(cr, uid, [('state', 'in', ['to upgrade', 'to remove', 'to install'])])
-        unmet_packages = []
-        mod_dep_obj = self.pool.get('ir.module.module.dependency')
-        for mod in mod_obj.browse(cr, uid, ids):
-            depends_mod_ids = mod_dep_obj.search(cr, uid, [('module_id', '=', mod.id)])
-            for dep_mod in mod_dep_obj.browse(cr, uid, depends_mod_ids):
-                if dep_mod.state in ('unknown','uninstalled'):
-                    unmet_packages.append(dep_mod.name)
-        if len(unmet_packages):
-            raise osv.except_osv(_('Unmet dependency !'), _('Following modules are not installed or unknown: %s') % ('\n\n' + '\n'.join(unmet_packages)))
-        mod_obj.download(cr, uid, ids, context=context)
-        cr.commit()
-        _db, pool = pooler.restart_pool(cr.dbname, update_module=True)
-
-        data_obj = pool.get('ir.model.data')
-        id2 = data_obj._get_id(cr, uid, 'base', 'view_base_module_upgrade_install')
-        if id2:
-            id2 = data_obj.browse(cr, uid, id2, context=context).res_id
+        ir_module = self.pool.get('ir.module.module')
+
+        # install/upgrade: double-check preconditions
+        ids = ir_module.search(cr, uid, [('state', 'in', ['to upgrade', 'to install'])])
+        if ids:
+            cr.execute("""SELECT d.name FROM ir_module_module m
+                                        JOIN ir_module_module_dependency d ON (m.id = d.module_id)
+                                        LEFT JOIN ir_module_module m2 ON (d.name = m2.name)
+                          WHERE m.id in %s and (m2.state IS NULL or m2.state IN %s)""",
+                      (tuple(ids), ('uninstalled',)))
+            unmet_packages = [x[0] for x in cr.fetchall()]
+            if unmet_packages:
+                raise osv.except_osv(_('Unmet dependency !'),
+                                     _('Following modules are not installed or unknown: %s') % ('\n\n' + '\n'.join(unmet_packages)))
 
+            ir_module.download(cr, uid, ids, context=context)
+            cr.commit() # save before re-creating cursor below
+
+        pooler.restart_pool(cr.dbname, update_module=True)
+
+        ir_model_data = self.pool.get('ir.model.data')
+        __, res_id = ir_model_data.get_object_reference(cr, uid, 'base', 'view_base_module_upgrade_install')
         return {
                 'view_type': 'form',
                 'view_mode': 'form',
                 'res_model': 'base.module.upgrade',
-                'views': [(id2, 'form')],
+                'views': [(res_id, 'form')],
                 'view_id': False,
                 'type': 'ir.actions.act_window',
                 'target': 'new',
@@ -115,4 +104,5 @@ class base_module_upgrade(osv.osv_memory):
     def config(self, cr, uid, ids, context=None):
         return self.pool.get('res.config').next(cr, uid, [], context=context)
 
-base_module_upgrade()
+
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: